Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
| 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
|