Update app.py

app.py

@@ -21,34 +21,34 @@ st.set_page_config(
 # Add all CSS styles
 st.markdown("""
     <style>
-        /* Progress Tracker Section */
-        .progress-step {
-            display: flex;
-            align-items: center;
-            margin: 10px 0;
-            font-size: 1.1em;
+        /* Progress Bar */
+        .progress-container {
+            width: 100%;
+            background: #f0f0f0;
+            border-radius: 8px;
+            margin: 20px 0;
         }
 
-        .progress-step.active::before {
-            content: '🍕';
-            margin-right: 10px;
-        }
-
-        .progress-step.done::before {
-            content: '✅';
-            margin-right: 10px;
+        .progress-bar {
+            height: 24px;
+            background: linear-gradient(90deg, #4caf50, #81c784);
+            border-radius: 8px;
+            text-align: center;
+            line-height: 24px;
+            color: white;
+            transition: width 0.5s ease;
         }
     </style>
-    
 """, unsafe_allow_html=True)
 
-def display_progress_tracker(container, steps, current_step_index):
-    """Display a progress tracker like a recipe being made."""
-    html = ""
-    for index, step in enumerate(steps):
-        step_class = "done" if index < current_step_index else "active" if index == current_step_index else ""
-        html += f"<div class='progress-step {step_class}'>{step}</div>"
-    container.markdown(html, unsafe_allow_html=True)
+def update_progress(progress_container, percentage, message=""):
+    """Update the progress bar."""
+    progress_html = f"""
+        <div class="progress-container">
+            <div class="progress-bar" style="width: {percentage}%">{message}</div>
+        </div>
+    """
+    progress_container.markdown(progress_html, unsafe_allow_html=True)
 
 def create_research_crew(topic: str):
     """Create the research crew with enhanced prompts"""
@@ -141,47 +141,33 @@ def create_research_crew(topic: str):
 def run_market_research(topic: str, progress_container, chat_container):
     """Run the market research process"""
     try:
-        steps = [
-            "Initializing research team...",
-            "Gathering market data...",
-            "Analyzing findings...",
-            "Compiling final report..."
-        ]
-        
         # Step 1: Initialize research team
-        display_progress_tracker(progress_container, steps, 0)
+        update_progress(progress_container, 0, "Initializing research team...")
         time.sleep(1)
         
         # Step 2: Gathering market data
-        display_progress_tracker(progress_container, steps, 1)
+        update_progress(progress_container, 25, "Gathering market data...")
         time.sleep(1)
         
         # Step 3: Analyzing findings
-        display_progress_tracker(progress_container, steps, 2)
+        update_progress(progress_container, 50, "Analyzing findings...")
         time.sleep(1)
         
         # Step 4: Compiling final report
-        display_progress_tracker(progress_container, steps, 3)
+        update_progress(progress_container, 75, "Compiling final report...")
         
         # Create and run the crew
         crew = create_research_crew(topic)
         result = crew.kickoff()
         
-        # Debugging: Display raw output to verify data
-        st.write("Debug: Raw output from CrewAI agents:")
-        st.write(result)
-        
         # Format and process the result
         report_data = format_json_output(result)
         
-        # Debugging: Display formatted JSON output
-        st.write("Debug: Formatted JSON output:")
-        st.write(report_data)
-        
         # Add visualizations
         charts = generate_visualizations(report_data)
         report_data['charts'] = charts
         
+        update_progress(progress_container, 100, "Report completed!")
         st.success("✨ Report generation completed! You can now view the full report.")
         
         # Store the report data in session state