Update app.py

app.py

@@ -54,53 +54,10 @@ chart_data_store = {}
 
 # Custom CSS for Finance theme with modern UI enhancements
 custom_css = """
-:root {
-    --main-bg: #ffffff;
-    --main-text: #000000;
-    --accent-primary: #2563eb;  // Bold blue
-    --accent-secondary: #dc2626; // Red for alerts
-    --accent-tertiary: #eab308;  // Yellow for highlights
-    --border-color: #e5e7eb;
-    --radius-md: 8px;
-    --spacing-sm: 8px;
-    --spacing-md: 16px;
-    --spacing-lg: 24px;
-    --shadow-sm: 0 1px 3px rgba(0,0,0,0.12);
-    --shadow-md: 0 4px 6px -1px rgba(0,0,0,0.1);
-    --shadow-lg: 0 10px 15px -3px rgba(0,0,0,0.1);
-}
-
-/* Chat Interface Styling */
-.chat-container {
-    background: var(--main-bg);
-    border-radius: var(--radius-md);
-    border: 1px solid var(--border-color);
-    padding: var(--spacing-md);
-    margin-bottom: var(--spacing-md);
-}
-
-.message {
-    padding: 12px 16px;
-    border-radius: 8px;
-    margin: 8px 0;
-    max-width: 80%;
-}
-
-.user-message {
-    background: #f0f4ff;
-    margin-left: auto;
-    border: 1px solid #d6e0ff;
-}
-
-.bot-message {
-    background: #fff;
-    margin-right: auto;
-    border: 1px solid #e0e0e0;
-}
-
+@import url('shared-base.css');
 /* Finance-specific styles */
 .finance-card {
-    background: #f8fafc;
+    background: var(--surface-2);
     border: 1px solid var(--border-color);
     border-radius: var(--radius-md);
     padding: var(--spacing-lg);
@@ -108,187 +65,45 @@ custom_css = """
     box-shadow: var(--shadow-md);
     transition: transform 0.3s, box-shadow 0.3s;
 }
-
-.header {
-    text-align: center;
-    padding: var(--spacing-lg) 0;
-    border-bottom: 1px solid var(--border-color);
-    margin-bottom: var(--spacing-lg);
-}
-
-.header-title {
-    font-size: 1.5rem;
-    font-weight: 600;
-    color: var(--main-text);
-    display: flex;
-    align-items: center;
-    justify-content: center;
-    gap: 8px;
-}
-
-.badge {
-    font-size: 0.7em;
-    padding: 2px 8px;
-    border-radius: 12px;
-    background: var(--accent-primary);
-    color: white;
-}
-
 .finance-card:hover {
     transform: translateY(-5px);
     box-shadow: var(--shadow-lg);
-    border-color: var(--accent-primary);
 }
-
 .chart-container {
-    background: var(--main-bg);
+    background: var(--white);
     border-radius: var(--radius-md);
     padding: var(--spacing-md);
     margin: var(--spacing-md) 0;
     box-shadow: var(--shadow-sm);
 }
-
 .news-feed {
     max-height: 400px;
     overflow-y: auto;
     padding: var(--spacing-md);
     border: 1px solid var(--border-color);
     border-radius: var(--radius-md);
-    background: var(--main-bg);
+    background: var(--surface-1);
 }
-
 .news-item {
     margin-bottom: var(--spacing-md);
     padding-bottom: var(--spacing-md);
     border-bottom: 1px solid var(--border-color);
 }
-
 .news-title {
     font-size: 1.1rem;
     font-weight: 600;
-    color: var(--main-text);
+    color: var(--text-color);
     margin-bottom: var(--spacing-sm);
 }
-
 .news-summary {
-    color: var(--main-text-secondary);
+    color: var(--text-color-secondary);
     line-height: 1.6;
 }
-
 .sentiment-gauge {
     width: 100%;
     height: 200px;
     margin: var(--spacing-md) 0;
 }
-
-/* Modern Financial Dashboard Styles */
-.dashboard-grid {
-    display: grid;
-    gap: 1.5rem;
-    grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
-}
-
-.ticker-tape {
-    background: linear-gradient(90deg, var(--accent-primary), var(--accent-secondary));
-    color: white;
-    padding: 1rem;
-    border-radius: var(--radius-md);
-    overflow: hidden;
-    position: relative;
-}
-
-.stock-card {
-    background: white;
-    border-radius: var(--radius-md);
-    padding: 1.5rem;
-    transition: transform 0.3s ease;
-    cursor: pointer;
-    position: relative;
-    overflow: hidden;
-}
-
-.stock-card::before {
-    content: '';
-    position: absolute;
-    top: -50%;
-    left: -50%;
-    width: 200%;
-    height: 200%;
-    background: linear-gradient(45deg, transparent, rgba(255,255,255,0.1));
-    transform: rotate(45deg);
-    transition: all 0.5s ease;
-}
-
-.stock-card:hover {
-    transform: translateY(-5px);
-    box-shadow: 0 10px 20px rgba(0,0,0,0.1);
-}
-
-.stock-card:hover::before {
-    transform: rotate(45deg) translateX(50%);
-}
-
-.realtime-chart {
-    border: 1px solid var(--border-color);
-    border-radius: var(--radius-md);
-    padding: 1rem;
-    background: white;
-    position: relative;
-}
-
-.chart-tooltip {
-    position: absolute;
-    background: var(--main-text);
-    color: var(--main-bg);
-    padding: 0.5rem 1rem;
-    border-radius: var(--radius-sm);
-    pointer-events: none;
-    opacity: 0;
-    transition: opacity 0.2s ease;
-}
-
-.dark-mode .ticker-tape {
-    background: linear-gradient(90deg, #1e3a8a, #991b1b) !important;
-}
-
-.dark-mode .stock-card {
-    background: #2d2d2d !important;
-}
-"""
-
-# Add JavaScript interactions
-financial_js = """
-<script>
-document.addEventListener('DOMContentLoaded', () => {
-    // Real-time chart interactions
-    const charts = document.querySelectorAll('.realtime-chart');
-    charts.forEach(chart => {
-        chart.addEventListener('mousemove', (e) => {
-            const tooltip = chart.querySelector('.chart-tooltip');
-            if(tooltip) {
-                tooltip.style.left = `${e.offsetX + 10}px`;
-                tooltip.style.top = `${e.offsetY + 10}px`;
-                tooltip.style.opacity = '1';
-            }
-        });
-        
-        chart.addEventListener('mouseleave', () => {
-            const tooltip = chart.querySelector('.chart-tooltip');
-            if(tooltip) tooltip.style.opacity = '0';
-        });
-    });
-
-    // Animated ticker tape
-    const ticker = document.querySelector('.ticker-tape');
-    if(ticker) {
-        let position = 0;
-        setInterval(() => {
-            position -= 1;
-            ticker.style.backgroundPosition = `${position}px 0`;
-        }, 50);
-    }
-});
-</script>
 """
 
 # Function to process PDF files
@@ -883,7 +698,6 @@ def analyze_ticker(ticker_input, period, use_brave_search=False):
         search_provider = "Brave Search" if (use_brave_search and BRAVE_API_KEY) else "Serper" if SERPER_API_KEY else "AI Knowledge Base"
         summary = f"""
 ### {ticker} Analysis (Using {search_provider})
-
 **Current Price:** ${stock_data['current_price']}  
 **52-Week High:** ${stock_data['52wk_high']}  
 **Market Cap:** ${stock_data['market_cap']:,}  
@@ -892,7 +706,6 @@ def analyze_ticker(ticker_input, period, use_brave_search=False):
 **Beta:** {stock_data['beta']}  
 **Avg Volume:** {stock_data['average_volume']:,}  
 {sentiment_summary}
-
 For in-depth analysis of this chart, ask the chatbot by typing "/chart" or "/analyze chart".
 For latest news, type "/news {ticker}".
         """
@@ -916,17 +729,14 @@ def analyze_image(image_file):
         mode = image.mode
         
         analysis = f"""## Technical Document Analysis
-
 **Image Properties:**
 - Dimensions: {width}x{height} pixels
 - Format: {format}
 - Color Mode: {mode}
-
 **Technical Analysis:**
 1. Document Quality:
    - Resolution: {'High' if width > 2000 or height > 2000 else 'Medium' if width > 1000 or height > 1000 else 'Low'}
    - Color Depth: {mode}
-
 2. Recommendations:
    - For text extraction, consider using PDF format
    - For technical diagrams, ensure high resolution
@@ -946,10 +756,8 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     
     gr.HTML("""
     <div class="header">
-        <div class="header-title">💰 Fin-Vision AI 
-            <span class="badge">Beta</span>
-        </div>
-        <div class="header-subtitle">Financial Analysis Copilot</div>
+        <div class="header-title">Fin-Vision AI</div>
+        <div class="header-subtitle">Advanced Financial Analysis Assistant</div>
     </div>
     """)
     
@@ -957,27 +765,35 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     with gr.Tabs() as main_tabs:
         # Chat Assistant Tab
         with gr.TabItem("💬 Chat Assistant", id=0):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    model_dropdown = gr.Dropdown(
+                        choices=["llama3-70b-8192", "llama3-8b-8192", "mixtral-8x7b-32768", "gemma-7b-it"],
+                        value="llama3-70b-8192",
+                        label="Model Selection"
+                    )
+                    web_search_toggle = gr.Checkbox(
+                        label="Enable Financial Search",
+                        value=True,
+                        info="Toggle web search functionality"
+                    )
+            
             chatbot = gr.Chatbot(
                 height=600,
                 show_copy_button=True,
                 elem_classes="chat-container",
                 container=True,
-                type="messages",
-                avatar_images=(
-                    "assets/user.png", 
-                    "assets/bot_finance.png"
-                )
+                type="messages"
             )
+            
             with gr.Row():
                 msg = gr.Textbox(
                     show_label=False,
-                    placeholder="💬 Ask about markets, type /news for updates, /chart for analysis...",
-                    scale=8,
-                    container=False
+                    placeholder="Ask about financial markets, analyze stocks, or discuss financial documents...",
+                    scale=8
                 )
-                with gr.Column(scale=1):
-                    send_btn = gr.Button("Send", variant="primary")
-                    clear_btn = gr.Button("🔄 Clear", variant="secondary")
+                send_btn = gr.Button("Send", scale=1)
+            clear_btn = gr.Button("Clear Conversation")
 
         # Document Analysis Tab
         with gr.TabItem("📄 Document Analysis", id=1):
@@ -1008,7 +824,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         with gr.TabItem("🔍 Financial Tools", id=2):
             with gr.Tabs() as financial_tabs:
                 # Stock Analysis
-                with gr.TabItem("📈 Stock Analysis"):
+                with gr.TabItem("Stock Analysis"):
                     with gr.Row():
                         with gr.Column():
                             ticker_input = gr.Textbox(
@@ -1026,7 +842,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                     stock_analysis = gr.Markdown()
 
                 # Market News
-                with gr.TabItem("📰 Market News"):
+                with gr.TabItem("Market News"):
                     news_ticker = gr.Textbox(
                         label="Company/Ticker",
                         placeholder="Enter company name or ticker symbol"
@@ -1035,7 +851,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                     news_results = gr.Markdown()
 
                 # Financial Report Analysis
-                with gr.TabItem("🖼️ Report Analysis"):
+                with gr.TabItem("Report Analysis"):
                     with gr.Row():
                         with gr.Column():
                             report_image = gr.File(