Update app.py

app.py

@@ -21,7 +21,6 @@ if not google_key or not tavily_key:
 # Enhanced CSS for a professional research interface
 CSS = """
 @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
-
 :root {
     --primary-color: #2563eb;
     --primary-hover: #1d4ed8;
@@ -36,19 +35,16 @@ CSS = """
     --warning-color: #f59e0b;
     --error-color: #ef4444;
 }
-
 body, .gradio-container {
     font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif !important;
     background-color: var(--bg-primary) !important;
     color: var(--text-primary) !important;
 }
-
 .gradio-container {
     max-width: 1200px !important;
     margin: 0 auto !important;
     padding: 2rem !important;
 }
-
 /* Header Styling */
 .header-container {
     text-align: center;
@@ -58,7 +54,6 @@ body, .gradio-container {
     border-radius: 16px;
     border: 1px solid var(--border-color);
 }
-
 h1 {
     font-size: 3rem;
     font-weight: 700;
@@ -67,13 +62,11 @@ h1 {
     -webkit-text-fill-color: transparent;
     margin-bottom: 0.5rem;
 }
-
 .subtitle {
     color: var(--text-secondary);
     font-size: 1.25rem;
     font-weight: 400;
 }
-
 /* Status Bar */
 .status-bar {
     background: var(--bg-secondary);
@@ -85,13 +78,11 @@ h1 {
     align-items: center;
     justify-content: space-between;
 }
-
 .status-indicator {
     display: flex;
     align-items: center;
     gap: 0.5rem;
 }
-
 .status-dot {
     width: 8px;
     height: 8px;
@@ -99,13 +90,11 @@ h1 {
     background: var(--success-color);
     animation: pulse 2s infinite;
 }
-
 @keyframes pulse {
     0% { opacity: 1; }
     50% { opacity: 0.5; }
     100% { opacity: 1; }
 }
-
 /* Chat Interface */
 #chatbot {
     background: var(--bg-secondary) !important;
@@ -113,21 +102,17 @@ h1 {
     border-radius: 16px !important;
     overflow: hidden !important;
 }
-
 #chatbot .message {
     border: none !important;
     padding: 1.5rem !important;
 }
-
 #chatbot .user {
     background: var(--bg-tertiary) !important;
     border-left: 4px solid var(--primary-color) !important;
 }
-
 #chatbot .bot {
     background: var(--bg-secondary) !important;
 }
-
 /* Progress Indicators */
 .progress-container {
     background: var(--bg-tertiary);
@@ -135,7 +120,6 @@ h1 {
     padding: 1rem;
     margin: 1rem 0;
 }
-
 .progress-bar {
     height: 4px;
     background: var(--border-color);
@@ -143,20 +127,17 @@ h1 {
     overflow: hidden;
     margin-top: 0.5rem;
 }
-
 .progress-fill {
     height: 100%;
     background: linear-gradient(90deg, var(--primary-color) 0%, #60a5fa 100%);
     transition: width 0.3s ease;
     animation: shimmer 2s infinite;
 }
-
 @keyframes shimmer {
     0% { opacity: 0.8; }
     50% { opacity: 1; }
     100% { opacity: 0.8; }
 }
-
 /* Input Area */
 .input-container {
     background: var(--bg-secondary);
@@ -165,7 +146,6 @@ h1 {
     padding: 1.5rem;
     margin-top: 2rem;
 }
-
 #chat-input textarea {
     background: var(--bg-tertiary) !important;
     color: var(--text-primary) !important;
@@ -175,12 +155,10 @@ h1 {
     font-size: 1rem !important;
     transition: all 0.2s ease !important;
 }
-
 #chat-input textarea:focus {
     border-color: var(--primary-color) !important;
     box-shadow: 0 0 0 3px rgba(37, 99, 235, 0.1) !important;
 }
-
 /* Buttons */
 .gr-button {
     background: var(--primary-color) !important;
@@ -192,22 +170,18 @@ h1 {
     transition: all 0.2s ease !important;
     cursor: pointer !important;
 }
-
 .gr-button:hover {
     background: var(--primary-hover) !important;
     transform: translateY(-1px);
     box-shadow: 0 4px 12px rgba(37, 99, 235, 0.3) !important;
 }
-
 .gr-button.secondary {
     background: var(--bg-tertiary) !important;
     color: var(--text-primary) !important;
 }
-
 .gr-button.secondary:hover {
     background: #475569 !important;
 }
-
 /* Report Display */
 .report-section {
     background: var(--bg-secondary);
@@ -216,7 +190,6 @@ h1 {
     padding: 2rem;
     margin: 1rem 0;
 }
-
 .report-section h2 {
     color: var(--text-primary);
     font-size: 1.75rem;
@@ -225,21 +198,18 @@ h1 {
     padding-bottom: 0.75rem;
     border-bottom: 2px solid var(--border-color);
 }
-
 .report-section h3 {
     color: var(--text-secondary);
     font-size: 1.25rem;
     font-weight: 500;
     margin: 1.5rem 0 0.75rem 0;
 }
-
 .source-list {
     background: var(--bg-tertiary);
     border-radius: 8px;
     padding: 1rem;
     margin-top: 1rem;
 }
-
 .source-item {
     display: flex;
     align-items: center;
@@ -249,11 +219,9 @@ h1 {
     text-decoration: none;
     transition: color 0.2s ease;
 }
-
 .source-item:hover {
     color: var(--primary-color);
 }
-
 /* Loading States */
 .thinking-indicator {
     display: flex;
@@ -262,12 +230,10 @@ h1 {
     color: var(--text-secondary);
     font-style: italic;
 }
-
 .thinking-dots {
     display: flex;
     gap: 0.25rem;
 }
-
 .thinking-dots span {
     width: 6px;
     height: 6px;
@@ -275,15 +241,12 @@ h1 {
     border-radius: 50%;
     animation: bounce 1.4s infinite ease-in-out both;
 }
-
 .thinking-dots span:nth-child(1) { animation-delay: -0.32s; }
 .thinking-dots span:nth-child(2) { animation-delay: -0.16s; }
-
 @keyframes bounce {
     0%, 80%, 100% { transform: scale(0); }
     40% { transform: scale(1); }
 }
-
 /* Export Options */
 .export-container {
     background: var(--bg-secondary);
@@ -292,13 +255,11 @@ h1 {
     padding: 1.5rem;
     margin-top: 2rem;
 }
-
 .export-buttons {
     display: flex;
     gap: 1rem;
     margin-top: 1rem;
 }
-
 /* Responsive Design */
 @media (max-width: 768px) {
     .gradio-container {
@@ -427,11 +388,8 @@ def chat_step(user_input, history, current_agent_state, topic_state, progress_st
         questions = get_clarifying_questions(planner_model, user_input)
         formatted_questions = f"""
 ### 🎯 Let's refine your research
-
 To create the most comprehensive report on **{user_input}**, I'd like to understand your specific interests:
-
 {questions}
-
 Please provide your answers below to help me tailor the research to your needs.
 """
         history[-1] = (user_input, formatted_questions)
@@ -446,16 +404,13 @@ Please provide your answers below to help me tailor the research to your needs.
             # Research and planning phase
             plan = research_and_plan(config, planner_model, tavily_client, topic_state, user_input)
             
-            # Show research plan
-            sections_preview = "\n".join([f"  {i+1}. {s.title}" for i, s in enumerate(plan['sections'])])
+            # Show research plan - FIXED: Safe access to section titles
+            sections_preview = "\n".join([f"  {i+1}. {s.get('title', f'Section {i+1}')}" for i, s in enumerate(plan['sections'])])
             planning_update = f"""
 ### 📊 Research Plan Created
-
 **Topic**: {plan['detailed_topic']}
-
 **Report Structure**:
 {sections_preview}
-
 Now conducting deep research and writing each section...
 """
             history[-1] = (user_input, planning_update)
@@ -482,14 +437,11 @@ Now conducting deep research and writing each section...
             # Final report display
             completion_message = f"""
 ### ✅ Research Complete!
-
 Your comprehensive research report is ready. You can:
 - 📥 Download as PDF using the button below
 - 📋 Copy the text directly from the report
 - 🔄 Start a new research topic
-
 ---
-
 {full_report}
 """
             history[-1] = (user_input, completion_message)