assaas

app.py

@@ -15,6 +15,7 @@ from modules.retriever import Retriever
 from modules.analyzer import Analyzer
 from modules.citation import CitationManager
 from modules.formatter import OutputFormatter
+from modules.visualizer import generate_chart, generate_line_chart, generate_pie_chart
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -617,6 +618,42 @@ html, body {
     color: #7e7e7e;
 }
 
+/* Accordion styling for collapsible sections */
+.accordion {
+    margin-bottom: 10px;
+    border: 1px solid #333333;
+    border-radius: 8px;
+    overflow: hidden;
+}
+
+.accordion-header {
+    background: #262626;
+    padding: 15px;
+    cursor: pointer;
+    font-weight: 600;
+    color: #00ffff;
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+}
+
+.accordion-header:hover {
+    background: #2d2d2d;
+}
+
+.accordion-content {
+    padding: 15px;
+    background: #1f1f1f;
+    border-top: 1px solid #333333;
+}
+
+.accordion-content h3 {
+    color: #00cccc !important;
+    margin-top: 0;
+    border-bottom: 1px solid #333333;
+    padding-bottom: 10px;
+}
+
 /* Spinner Animation */
 @keyframes spin {
     0% { transform: rotate(0deg); }
@@ -626,110 +663,82 @@ html, body {
 /* Responsive Design */
 @media (max-width: 768px) {
     .title {
-        font-size: 2rem;
+        font-size: 1.8rem;
     }
 
     .subtitle {
-        font-size: 1.1rem;
+        font-size: 1rem;
     }
 
     .card {
-        padding: 20px 15px;
+        padding: 15px;
     }
 
     .btn-primary {
-        padding: 12px 20px;
-        font-size: 16px;
+        font-size: 15px;
+        padding: 12px;
     }
 
     .streaming-content {
-        padding: 15px;
-        font-size: 15px;
+        font-size: 14px;
+        padding: 10px;
+    }
+
+    .gr-textbox input, .gr-textbox textarea {
+        font-size: 14px;
+    }
+
+    .accordion {
+        font-size: 14px;
+    }
+
+    .container {
+        padding: 10px;
     }
 }
-"""
 
-def research_assistant(query, file, use_ddg, progress=gr.Progress()):
-    """Main entry point for the research assistant with streaming"""
-    logging.info(f"Research assistant called with query: {query}")
-    for step in orchestrator.run(query, file, use_ddg, progress):
-        yield step
-
-# Create Gradio interface
-with gr.Blocks(
-    css=custom_css, 
-    title="Research Assistant",
-    head="""
-        <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.9.0/styles/github-dark.min.css">
-        <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.9.0/highlight.min.js"></script>
-        <script>hljs.highlightAll();</script>
-    """
-) as demo:
-    gr.Markdown(f"""
-    <div class="header">
-        <h1 class="title">🧠 AI Research Assistant</h1>
-        <p class="subtitle">Your intelligent research companion</p>
-        <div class="version-info">🧬 Version: {VERSION['version']}</div>
-    </div>
-    """)
+/* Mobile optimization */
+@media (max-width: 480px) {
+    .header {
+        padding: 15px 10px;
+    }
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            with gr.Group(elem_classes=["card", "input-section"]):
-                gr.Markdown("### 🔍 Research Input")
-                query_input = gr.Textbox(
-                    label="Research Query",
-                    placeholder="Enter your research question...",
-                    lines=3
-                )
-                file_upload = gr.File(
-                    label="📄 Upload Files (PDF, Images)",
-                    file_types=[".pdf", ".png", ".jpg", ".jpeg"]
-                )
-                ddg_checkbox = gr.Checkbox(
-                    label="Use DuckDuckGo Search (Alternative)",
-                    value=False
-                )
-                submit_btn = gr.Button("Research", elem_classes=["btn-primary"])
-        
-        with gr.Column(scale=2):
-            with gr.Group(elem_classes=["card", "output-section"]):
-                gr.Markdown("### 📊 Analysis Results")
-                
-                # Progress bar
-                progress_container = gr.HTML("""
-                    <div class="progress-container">
-                        <div class="progress-bar" id="progress-bar"></div>
-                    </div>
-                """)
-                
-                output = gr.Markdown(elem_classes=["streaming-content"], visible=True)
-                status_indicator = gr.HTML("""
-                <div class="status-indicator">
-                    <div class="spinner"></div>
-                    <span class="progress-text">Ready for your research query</span>
-                </div>
-                """)
+    .title {
+        font-size: 1.5rem;
+    }
+    
+    .card {
+        padding: 12px;
+    }
     
-    submit_btn.click(
-        fn=research_assistant,
-        inputs=[query_input, file_upload, ddg_checkbox],
-        outputs=output
-    )
+    .btn-primary {
+        padding: 10px;
+        font-size: 14px;
+    }
+}
+"""
+
+def parse_analysis_sections(analysis_text):
+    """Parse analysis text into sections"""
+    sections = {
+        "Overview": "",
+        "Key Findings": "",
+        "Perspectives": "",
+        "Implications": "",
+        "Controversies": ""
+    }
     
-    query_input.submit(
-        fn=research_assistant,
-        inputs=[query_input, file_upload, ddg_checkbox],
-        outputs=output
-    )
+    current_section = None
+    lines = analysis_text.split('\n')
     
-    gr.Markdown("""
-    <div class="footer">
-        <p>Built with ❤️ for researchers worldwide</p>
-        <p><span class="highlight">Features:</span> File upload • Web search • Real-time analysis</p>
-        <p><strong>Note:</strong> The AI model server may take up to 4-5 minutes to start initially.</p>
-    </div>
-    """)
-
-if __name__ == "__main__":
-    demo.launch(share=True)
+    for line in lines:
+        if line.startswith("# Overview"):
+            current_section = "Overview"
+            continue
+        elif line.startswith("# Key Findings"):
+            current_section = "Key Findings"
+            continue
+        elif line.startswith("# Perspectives"):
+            current_section = "Perspectives"
+            continue
+        elif line.startswith

modules/analyzer.py

@@ -104,7 +104,20 @@ Please provide:
 4. Potential implications or future directions
 5. Any controversies or conflicting viewpoints
 
-Structure your response clearly with these sections. If there is insufficient information, state that clearly."""
+Structure your response clearly with these sections. If there is insufficient information, state that clearly.
+
+Format:
+- Use the exact headings below
+- Use markdown for formatting
+- Wrap code in triple backticks
+- Separate each section clearly
+
+Sections:
+# Overview
+# Key Findings
+# Perspectives
+# Implications
+# Controversies"""
         
         try:
             # First check if server is ready with extended timeout and user feedback
@@ -175,7 +188,20 @@ Please provide:
 4. Potential implications or future directions
 5. Any controversies or conflicting viewpoints
 
-Structure your response clearly with these sections. If there is insufficient information, state that clearly."""
+Structure your response clearly with these sections. If there is insufficient information, state that clearly.
+
+Format:
+- Use the exact headings below
+- Use markdown for formatting
+- Wrap code in triple backticks
+- Separate each section clearly
+
+Sections:
+# Overview
+# Key Findings
+# Perspectives
+# Implications
+# Controversies"""
         
         try:
             # First check if server is ready with extended timeout

modules/visualizer.py

@@ -1,111 +1,50 @@
 # modules/visualizer.py
-import pandas as pd
+import base64
+from io import BytesIO
 import matplotlib.pyplot as plt
-from datetime import datetime
-import os
-import logging
+import pandas as pd
+import numpy as np
 
-def load_status_log(log_file="server_status_log.csv"):
-    """Load server status log from CSV file"""
-    if not os.path.exists(log_file):
-        raise FileNotFoundError(f"Log file not found: {log_file}")
+def generate_chart(data, title="Chart", xlabel="X", ylabel="Y"):
+    """Generate a chart and return as base64 encoded string"""
+    plt.figure(figsize=(8, 4))
+    plt.bar(data.keys(), data.values())
+    plt.title(title)
+    plt.xlabel(xlabel)
+    plt.ylabel(ylabel)
+    plt.tight_layout()
     
-    try:
-        df = pd.read_csv(log_file, parse_dates=["timestamp"])
-        return df
-    except Exception as e:
-        logging.error(f"Error loading log file: {e}")
-        raise
-
-def plot_uptime_trend(df, output_file="server_uptime_trend.png"):
-    """Plot server uptime trend over time"""
-    try:
-        # Sort by timestamp
-        df = df.sort_values("timestamp")
-        
-        # Convert status to boolean
-        df["status_bool"] = df["status"].map({"UP": 1, "DOWN": 0})
-        
-        plt.figure(figsize=(12, 4))
-        plt.plot(df["timestamp"], df["status_bool"], drawstyle='steps-post', 
-                marker='o', linestyle='-', markersize=3, color='#1f77b4')
-        
-        plt.title("Server Uptime Trend", fontsize=14, pad=20)
-        plt.xlabel("Time", fontsize=12)
-        plt.ylabel("Status", fontsize=12)
-        plt.yticks([0, 1], ["DOWN", "UP"])
-        plt.grid(True, linestyle='--', alpha=0.5)
-        plt.tight_layout()
-        plt.savefig(output_file)
-        plt.close()
-        
-        print(f"📈 Chart saved to: {output_file}")
-        logging.info(f"Uptime trend chart saved to: {output_file}")
-        return output_file
-    except Exception as e:
-        logging.error(f"Error plotting uptime trend: {e}")
-        raise
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    plt.close()
+    data_uri = base64.b64encode(buf.getvalue()).decode('utf-8')
+    return f"![{title}](data:image/png;base64,{data_uri})"
 
-def plot_uptime_summary(df, output_file="server_uptime_summary.png"):
-    """Plot server uptime summary as pie chart"""
-    try:
-        # Calculate uptime percentage
-        total = len(df)
-        up = df[df["status"] == "UP"].shape[0]
-        down = df[df["status"] == "DOWN"].shape[0]
-        
-        if total == 0:
-            raise ValueError("No data in log file")
-        
-        uptime_pct = (up / total) * 100
-        downtime_pct = (down / total) * 100
-        
-        labels = [f'UP ({uptime_pct:.1f}%)', f'DOWN ({downtime_pct:.1f}%)']
-        sizes = [up, down]
-        colors = ['#28a745', '#dc3545']
-        
-        plt.figure(figsize=(6, 6))
-        wedges, texts, autotexts = plt.pie(sizes, labels=labels, autopct='%1.1f%%', 
-                                          startangle=140, colors=colors)
-        
-        # Make percentage text white and bold
-        for autotext in autotexts:
-            autotext.set_color('white')
-            autotext.set_fontweight('bold')
-        
-        plt.title("Server Uptime Summary", fontsize=14, pad=20)
-        plt.axis('equal')
-        plt.tight_layout()
-        plt.savefig(output_file)
-        plt.close()
-        
-        print(f"📊 Summary chart saved to: {output_file}")
-        logging.info(f"Uptime summary chart saved to: {output_file}")
-        return output_file
-    except Exception as e:
-        logging.error(f"Error plotting uptime summary: {e}")
-        raise
+def generate_line_chart(data, title="Line Chart", xlabel="X", ylabel="Y"):
+    """Generate a line chart and return as base64 encoded string"""
+    plt.figure(figsize=(8, 4))
+    plt.plot(list(data.keys()), list(data.values()), marker='o')
+    plt.title(title)
+    plt.xlabel(xlabel)
+    plt.ylabel(ylabel)
+    plt.grid(True)
+    plt.tight_layout()
+    
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    plt.close()
+    data_uri = base64.b64encode(buf.getvalue()).decode('utf-8')
+    return f"![{title}](data:image/png;base64,{data_uri})"
 
-def get_uptime_stats(df):
-    """Get uptime statistics"""
-    try:
-        total = len(df)
-        up = df[df["status"] == "UP"].shape[0]
-        down = df[df["status"] == "DOWN"].shape[0]
-        
-        if total == 0:
-            return {"total_checks": 0, "uptime_pct": 0, "downtime_pct": 0}
-        
-        uptime_pct = (up / total) * 100
-        downtime_pct = (down / total) * 100
-        
-        return {
-            "total_checks": total,
-            "up_checks": up,
-            "down_checks": down,
-            "uptime_pct": round(uptime_pct, 2),
-            "downtime_pct": round(downtime_pct, 2)
-        }
-    except Exception as e:
-        logging.error(f"Error calculating uptime stats: {e}")
-        raise
+def generate_pie_chart(data, title="Pie Chart"):
+    """Generate a pie chart and return as base64 encoded string"""
+    plt.figure(figsize=(6, 6))
+    plt.pie(data.values(), labels=data.keys(), autopct='%1.1f%%')
+    plt.title(title)
+    plt.tight_layout()
+    
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    plt.close()
+    data_uri = base64.b64encode(buf.getvalue()).decode('utf-8')
+    return f"![{title}](data:image/png;base64,{data_uri})"