change pdf image width

microsoft · Jul 31, 2024 · a3bffa6 · a3bffa6
1 parent affc228
commit a3bffa6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/rdagent/log/ui/app.py b/rdagent/log/ui/app.py
@@ -354,9 +354,9 @@ def tasks_window(tasks: list[FactorTask | ModelTask]):
 else:
     round = 1
 
-rf_c, d_c = st.columns([2, 2])
 
 # Research & Feedback Window
+rf_c, d_c = st.columns([2, 2])
 with rf_c:
     if state.log_type in ["qlib_model", "qlib_factor"]:
         # Research Window
@@ -365,7 +365,7 @@ def tasks_window(tasks: list[FactorTask | ModelTask]):
             # pdf image
             if pim := state.msgs[round]["r.extract_factors_and_implement.load_pdf_screenshot"]:
                 for i in range(min(2, len(pim))):
-                    st.image(pim[i].content)
+                    st.image(pim[i].content, width=200)
 
             # Hypothesis
             if hg := state.msgs[round]["r.hypothesis generation"]:
@@ -406,7 +406,7 @@ def tasks_window(tasks: list[FactorTask | ModelTask]):
             st.subheader("Research🔍", divider="blue", anchor="_research")
             if pim := state.msgs[round]["r.pdf_image"]:
                 for i in range(len(pim)):
-                    st.image(pim[i].content)
+                    st.image(pim[i].content, width=200)
 
             # loaded model exp
             if mem := state.msgs[round]["d.load_experiment"]: