Update app.py

app.py

@@ -1,13 +1,14 @@
-# app.py
+
 import os
 import time
 import json
 import random
+import re
+
+import plotly.graph_objects as go
 import streamlit as st
-from dotenv import load_dotenv
 from crewai import Agent, Crew, Process, Task
-import plotly.graph_objects as go
-import re
+from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
@@ -141,8 +142,10 @@ st.markdown("""
             background: linear-gradient(
                 90deg,
                 transparent,
-                rgba(255, 255, 255, 0.4),
-                transparent
+                rgba(255, 255, 255,   
+ 0.4),
+                transparent   
+
             );
             animation: shimmer 2s infinite;
         }
@@ -184,6 +187,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
+
 def format_json_output(raw_output):
     """Format CrewOutput or raw string into proper JSON structure"""
     try:
@@ -199,8 +203,8 @@ def format_json_output(raw_output):
         if match:
             try:
                 return json.loads(match.group())
-            except:
-                pass
+            except json.JSONDecodeError:
+                pass  # Or handle the error as needed
 
         # If no JSON found, create structured format
         return {
@@ -208,7 +212,16 @@ def format_json_output(raw_output):
             "detailed_report": raw_text,
             "sources": extract_sources(raw_text)
         }
-        
+
+    except Exception as e:
+        st.error(f"Error formatting output: {str(e)}")
+        return {
+            "exec_summary": "Error formatting report",
+            "detailed_report": raw_text if 'raw_text' in locals() else str(raw_output),
+            "sources": []
+        }
+
+
 def display_thinking_animation():
     """Display thinking animation dots"""
     return """
@@ -219,6 +232,7 @@ def display_thinking_animation():
         </div>
     """
 
+
 def display_agent_message(container, agent_type: str, message: str, thinking: bool = False):
     """Display an animated agent message"""
     icons = {
@@ -226,7 +240,7 @@ def display_agent_message(container, agent_type: str, message: str, thinking: bo
         "analyst": "📊",
         "writer": "✍️"
     }
-    
+
     message_html = f"""
         <div class="agent-message">
             <div class="agent-avatar {agent_type}">
@@ -240,6 +254,7 @@ def display_agent_message(container, agent_type: str, message: str, thinking: bo
     """
     container.markdown(message_html, unsafe_allow_html=True)
 
+
 def update_progress(container, progress, message=""):
     """Update progress bar with animation"""
     progress_html = f"""
@@ -252,6 +267,7 @@ def update_progress(container, progress, message=""):
     """
     container.markdown(progress_html, unsafe_allow_html=True)
 
+
 def run_market_research(topic: str, progress_container, chat_container):
     try:
         researcher = Agent(
@@ -260,14 +276,14 @@ def run_market_research(topic: str, progress_container, chat_container):
             backstory='You are an experienced market research analyst with expertise in data analysis and trend identification.',
             verbose=True
         )
-        
+
         analyst = Agent(
             role='Data Analyst',
             goal='Create data-driven insights with specific metrics',
             backstory='You are a skilled data analyst who specializes in turning research into actionable insights.',
             verbose=True
         )
-        
+
         writer = Agent(
             role='Report Writer',
             goal='Create both executive summary and detailed reports with citations',
@@ -276,14 +292,14 @@ def run_market_research(topic: str, progress_container, chat_container):
         )
 
         # Research Phase
-        display_agent_message(chat_container, "researcher", 
-            f"👋 Hello! I'll be researching the {topic} market thoroughly.", False)
+        display_agent_message(chat_container, "researcher",
+                              f"👋 Hello! I'll be researching the {topic} market thoroughly.", False)
         time.sleep(1)
-        
+
         update_progress(progress_container, 20, "🔍 Gathering market data...")
         display_agent_message(chat_container, "researcher", "Processing market information...", True)
         time.sleep(1)
-        
+
         research_task = Task(
             description=f"""
             Research {topic} with focus on:
@@ -292,7 +308,7 @@ def run_market_research(topic: str, progress_container, chat_container):
             3. Competitive analysis
             4. Future trends
             5. Include verifiable sources
-            
+
             Format the output as clear sections with numerical data points.
             """,
             agent=researcher,
@@ -301,10 +317,10 @@ def run_market_research(topic: str, progress_container, chat_container):
 
         # Analysis Phase
         update_progress(progress_container, 40, "📊 Analyzing findings...")
-        display_agent_message(chat_container, "analyst", 
-            "I've received the research data. Beginning analysis...", False)
+        display_agent_message(chat_container, "analyst",
+                              "I've received the research data. Beginning analysis...", False)
         time.sleep(1)
-        
+
         analysis_task = Task(
             description=f"""
             Analyze the research findings and provide:
@@ -312,7 +328,7 @@ def run_market_research(topic: str, progress_container, chat_container):
             2. Market share analysis
             3. Competitive landscape
             4. Key opportunities and challenges
-            
+
             Include specific metrics and percentages.
             """,
             agent=analyst,
@@ -322,17 +338,17 @@ def run_market_research(topic: str, progress_container, chat_container):
 
         # Report Phase
         update_progress(progress_container, 70, "✍️ Generating report...")
-        display_agent_message(chat_container, "writer", 
-            "Converting analysis into comprehensive report...", False)
+        display_agent_message(chat_container, "writer",
+                              "Converting analysis into comprehensive report...", False)
         time.sleep(1)
-        
+
         report_task = Task(
             description=f"""
             Create a market research report with:
             1. Executive Summary (2-3 paragraphs)
             2. Detailed Report (full analysis)
             3. Sources and Citations
-            
+
             Format as a JSON with:
             {{
                 "exec_summary": "summary text",
@@ -353,32 +369,31 @@ def run_market_research(topic: str, progress_container, chat_container):
         )
 
         result = crew.kickoff()
-        
+
         # Update final progress
         update_progress(progress_container, 100, "✨ Report completed!")
-        display_agent_message(chat_container, "writer", 
-            "Report generation completed! You can now view the full report.", False)
+        display_agent_message(chat_container, "writer",
+                              "Report generation completed! You can now view the full report.", False)
 
         return format_json_output(result)
 
     except Exception as e:
-        st.error(f"Error formatting output: {str(e)}")
+        st.error(f"Error generating report: {str(e)}")
         return {
-            "exec_summary": "Error formatting report",
-            "detailed_report": raw_text if 'raw_text' in locals() else str(raw_output),
+            "exec_summary": "Error generating report",
+            "detailed_report": "",
             "sources": []
         }
 
+
 def extract_section(text, section_name):
     """Extract a section from the text"""
-    pattern = f"{section_name}.*?\n(.*?)(?=\n\n|$)"
-    match = re.search(pattern, text, re.DOTALL | re.IGNORECASE)
-    return match.group(1).strip() if match else ""
-
-def extract_sources(text):
-    """Extract sources from the text"""
-    sources = []
-    source_pattern = r"Source:.*?(?:\n|$)|\[.*?\]|\(https?://.*?\)"
+    pattern = f"{section_name}.*?\n(.*?)(?=\n\n|<span class="math-inline">\)"
+match \= re\.search\(pattern, text, re\.DOTALL \| re\.IGNORECASE\)
+return match\.group\(1\)\.strip\(\) if match else ""
+def extract\_sources\(text\)\:
+"""Extract sources from the text"""
+source\_pattern \= r"Source\:\.\*?\(?\:\\n\|</span>)|\[.*?\]|\(https?://.*?\)"
     matches = re.finditer(source_pattern, text, re.MULTILINE)
     return [match.group().strip() for match in matches]