QuantumLearner commited on
Commit
f5da570
·
verified ·
1 Parent(s): 1fa4a19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -23,9 +23,13 @@ if not openai_api_key or not tavily_api_key:
23
  async def get_report(query: str, report_type: str, sources: list, report_source: str):
24
  f = io.StringIO()
25
  logs_container = st.empty()
 
 
 
 
 
26
  with redirect_stdout(f):
27
  if report_source == 'local':
28
- # Set the DOC_PATH environment variable
29
  os.environ['DOC_PATH'] = './uploads'
30
  researcher = GPTResearcher(query=query, report_type=report_type, report_source=report_source)
31
  else:
@@ -35,7 +39,8 @@ async def get_report(query: str, report_type: str, sources: list, report_source:
35
 
36
  while True:
37
  logs = f.getvalue()
38
- logs_container.text_area("Agent Logs", logs, height=200, key="realtime_logs_display")
 
39
  await asyncio.sleep(1) # Update every second
40
  if "Finalized research step" in logs:
41
  break
 
23
  async def get_report(query: str, report_type: str, sources: list, report_source: str):
24
  f = io.StringIO()
25
  logs_container = st.empty()
26
+ logs = ''
27
+
28
+ # Create the text area widget once before the loop
29
+ logs_container.text_area("Agent Logs", logs, height=200, key="realtime_logs_display")
30
+
31
  with redirect_stdout(f):
32
  if report_source == 'local':
 
33
  os.environ['DOC_PATH'] = './uploads'
34
  researcher = GPTResearcher(query=query, report_type=report_type, report_source=report_source)
35
  else:
 
39
 
40
  while True:
41
  logs = f.getvalue()
42
+ # Update the content of the text area via session_state
43
+ st.session_state.realtime_logs_display = logs
44
  await asyncio.sleep(1) # Update every second
45
  if "Finalized research step" in logs:
46
  break