Update app.py

app.py

@@ -3,18 +3,23 @@ from dotenv import load_dotenv
 import os
 from agents import create_research_crew
 from utils import process_crew_output, display_enhanced_report, update_progress
-from huggingface_hub import HfApi
-from huggingface_hub.inference._client import InferenceClient
+import queue
+import threading
 
 load_dotenv()
 
-st.set_page_config(
-    page_title="Market Research Generator",
-    page_icon="📊",
-    layout="wide"
-)
+# Initialize queue for logs
+if 'log_queue' not in st.session_state:
+    st.session_state.log_queue = queue.Queue()
+
+def init_session_state():
+    if 'report' not in st.session_state:
+        st.session_state.report = None
+    if 'generating' not in st.session_state:
+        st.session_state.generating = False
+    if 'logs' not in st.session_state:
+        st.session_state.logs = []
 
-# Get API keys from Hugging Face secrets
 def get_api_keys():
     try:
         openai_key = st.secrets["OPENAI_API_KEY"]
@@ -24,11 +29,8 @@ def get_api_keys():
         st.error(f"Error loading API keys: {str(e)}")
         return None, None
 
-def init_session_state():
-    if 'report' not in st.session_state:
-        st.session_state.report = None
-    if 'generating' not in st.session_state:
-        st.session_state.generating = False
+def log_message(message):
+    st.session_state.log_queue.put(message)
 
 def run_market_research(topic: str, questions: list, progress_container):
     try:
@@ -62,7 +64,7 @@ def main():
     os.environ["OPENAI_API_KEY"] = openai_key
     os.environ["SERPER_API_KEY"] = serper_key
     
-    tab1, tab2 = st.tabs(["Generate Report", "View Report"])
+    tab1, tab2, tab3 = st.tabs(["Generate Report", "View Report", "Live Logs"])
     
     with tab1:
         topic = st.text_input("Market Research Topic", 
@@ -84,6 +86,9 @@ def main():
             with st.spinner("Generating comprehensive report..."):
                 st.session_state.generating = True
                 try:
+                    # Clear previous logs
+                    st.session_state.logs = []
+                    
                     report = run_market_research(
                         topic=topic,
                         questions=questions.split('\n'),
@@ -99,6 +104,19 @@ def main():
         if st.session_state.report:
             st.warning("⚠️ AI-generated report. Verify critical information independently.")
             display_enhanced_report(st.session_state.report)
+    
+    with tab3:
+        st.subheader("Live Generation Logs")
+        log_container = st.empty()
+        
+        # Display logs
+        while not st.session_state.log_queue.empty():
+            log = st.session_state.log_queue.get()
+            st.session_state.logs.append(log)
+        
+        # Show all logs
+        for log in st.session_state.logs:
+            log_container.markdown(log)
 
 if __name__ == "__main__":
     main()