Update app.py

app.py

@@ -326,18 +326,6 @@ Question: {query} [/INST]</s>"""
         self.conversation_history = []
         return "Conversation has been reset."
 
-# Main function with Gradio UI
-import os
-import re
-import json
-import time
-from tqdm import tqdm
-from pathlib import Path
-import spaces
-import gradio as gr
-
-# WARNING: Don't import torch, cuda, or GPU-related modules at the top level
-# All previous helper functions and classes remain the same
 
 def main():
     # Create the Vision 2030 service
@@ -377,16 +365,23 @@ def main():
         .chat-input { background-color: white; border-radius: 8px; border: 1px solid #ddd; }
         .info-box { background-color: #f8f9fa; padding: 10px; border-radius: 8px; margin-top: 10px; }
     """) as demo:
-        # Header with logo and title
+        # Header with stylized title (no external images)
         with gr.Row():
-            with gr.Column(scale=1):
-                gr.Image("https://www.vision2030.gov.sa/media/g0dp5h3t/vision-2030-logo-en.png", 
-                         show_label=False, container=False, height=100, 
-                         elem_classes=["header-img"])
-            with gr.Column(scale=3):
-                gr.Markdown("""
-                # Vision 2030 Assistant
-                ### Your interactive guide to Saudi Arabia's national transformation program
+            with gr.Column():
+                gr.HTML("""
+                <div style="display: flex; align-items: center; margin-bottom: 20px;">
+                    <div style="background: linear-gradient(135deg, #1e9e5a, #45b08c); 
+                         color: white; padding: 15px; border-radius: 10px; 
+                         margin-right: 20px; width: 80px; height: 80px; 
+                         display: flex; justify-content: center; align-items: center; 
+                         font-weight: bold; font-size: 24px;">
+                        V2030
+                    </div>
+                    <div>
+                        <h1 style="margin: 0; color: #1e9e5a;">Vision 2030 Assistant</h1>
+                        <h3 style="margin: 5px 0 0 0; font-weight: normal;">Your interactive guide to Saudi Arabia's national transformation program</h3>
+                    </div>
+                </div>
                 """)
             
         # Language toggle in the header
@@ -408,10 +403,6 @@ def main():
                         chatbot = gr.Chatbot(
                             height=450,
                             bubble_full_width=False,
-                            avatar_images=(
-                                "https://api.iconify.design/fluent-emoji:person.svg",
-                                "https://api.iconify.design/fluent-emoji:robot.svg"
-                            ),
                             show_label=False
                         )