Update app.py

app.py

@@ -40,6 +40,12 @@ class ConversationMemory:
         }
         self.conversations.append(interaction)
     
+    def clear_history(self):
+        """Clear conversation history"""
+        self.conversations = []
+        self.context_graph = {}
+        self.session_data = {}
+    
     def get_relevant_context(self, query: str, limit: int = 3) -> List[Dict]:
         # Simple relevance scoring - in production, use embeddings
         relevant = []
@@ -79,9 +85,9 @@ class NexusAI:
         try:
             self.client = Together(api_key=api_key)
             self.api_key = api_key
-            return True, "API key initialized successfully!"
+            return True, "✅ API key initialized successfully!"
         except Exception as e:
-            return False, f"Failed to initialize API key: {str(e)}"
+            return False, f"❌ Failed to initialize API key: {str(e)}"
     
     def extract_text_from_pdf(self, pdf_path: str) -> str:
         """Extract text from PDF file"""
@@ -249,8 +255,8 @@ def process_pdf_input(pdf_file, user_question: str, api_key_status: str) -> str:
     if pdf_file is None:
         return "Please upload a PDF file first!"
     
-    # Extract text from PDF
-    extracted_text = nexus_ai.extract_text_from_pdf(pdf_file.name)
+    # Extract text from PDF - pdf_file is already a file path string
+    extracted_text = nexus_ai.extract_text_from_pdf(pdf_file)
     
     if user_question.strip():
         return nexus_ai.generate_response(user_question, "pdf", extracted_text)
@@ -265,8 +271,8 @@ def process_image_input(image_file, user_question: str, api_key_status: str) ->
     if image_file is None:
         return "Please upload an image file first!"
     
-    # Analyze image
-    image_analysis = nexus_ai.analyze_image(image_file.name)
+    # Analyze image - image_file is already a file path string
+    image_analysis = nexus_ai.analyze_image(image_file)
     
     if user_question.strip():
         return nexus_ai.generate_response(user_question, "image", image_analysis)
@@ -281,8 +287,8 @@ def process_audio_input(audio_file, user_question: str, api_key_status: str) ->
     if audio_file is None:
         return "Please upload an audio file first!"
     
-    # Transcribe audio
-    transcribed_text = nexus_ai.transcribe_audio(audio_file.name)
+    # Transcribe audio - audio_file is already a file path string
+    transcribed_text = nexus_ai.transcribe_audio(audio_file)
     
     if user_question.strip():
         combined_input = f"Transcribed audio: '{transcribed_text}'\n\nUser question: {user_question}"
@@ -318,7 +324,7 @@ def show_conversation_history() -> str:
         return "No conversation history yet. Start chatting to build your knowledge base!"
     
     history = "## 📚 Recent Conversation History\n\n"
-    for i, conv in enumerate(nexus_ai.memory.conversations[-5:], 1):  # Show last 5
+    for i, conv in enumerate(nexus_ai.memory.conversations[-10:], 1):  # Show last 10
         timestamp = datetime.fromisoformat(conv["timestamp"]).strftime("%H:%M:%S")
         history += f"**{i}. [{conv['input_type'].upper()}] {timestamp}**\n"
         history += f"Input: {conv['content'][:100]}{'...' if len(conv['content']) > 100 else ''}\n"
@@ -326,49 +332,168 @@ def show_conversation_history() -> str:
     
     return history
 
+def clear_conversation_history() -> str:
+    """Clear conversation history"""
+    nexus_ai.memory.clear_history()
+    return "✅ Conversation history has been cleared!"
+
 # Create the Gradio interface
 def create_nexus_interface():
     with gr.Blocks(
         theme=gr.themes.Soft(),
         title="Nexus AI Assistant",
         css="""
+        /* Center the main container */
         .gradio-container {
-            max-width: 1200px !important;
+            max-width: 1400px !important;
+            margin: 0 auto !important;
+            padding: 20px !important;
+        }
+        
+        /* API Key section styling */
+        .api-key-section {
+            background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
+            border-radius: 12px;
+            padding: 20px;
+            margin-bottom: 25px;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.07);
+            border: 1px solid #e1e8ed;
+        }
+        
+        /* Button styling */
+        .primary-button {
+            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+            border: none !important;
+            color: white !important;
+            font-weight: 600 !important;
+            border-radius: 8px !important;
+            padding: 12px 24px !important;
+            transition: all 0.3s ease !important;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1) !important;
+        }
+        
+        .primary-button:hover {
+            transform: translateY(-2px) !important;
+            box-shadow: 0 4px 12px rgba(0, 0, 0, 0.2) !important;
+        }
+        
+        .secondary-button {
+            background: linear-gradient(135deg, #74b9ff 0%, #0984e3 100%) !important;
+            border: none !important;
+            color: white !important;
+            font-weight: 500 !important;
+            border-radius: 8px !important;
+            transition: all 0.3s ease !important;
+        }
+        
+        .danger-button {
+            background: linear-gradient(135deg, #fd79a8 0%, #e84393 100%) !important;
+            border: none !important;
+            color: white !important;
+            font-weight: 500 !important;
+            border-radius: 8px !important;
+            transition: all 0.3s ease !important;
+        }
+        
+        /* Tab styling */
+        .tab-nav button {
+            border-radius: 8px 8px 0 0 !important;
+            font-weight: 500 !important;
+            padding: 12px 20px !important;
+        }
+        
+        /* Text area with scrollbar */
+        .scrollable-textarea textarea {
+            overflow-y: auto !important;
+            resize: vertical !important;
+        }
+        
+        /* Card styling for better visual separation */
+        .input-card {
+            background: #ffffff;
+            border-radius: 10px;
+            padding: 20px;
+            box-shadow: 0 2px 8px rgba(0, 0, 0, 0.05);
+            border: 1px solid #f0f0f0;
+            margin-bottom: 15px;
+        }
+        
+        .output-card {
+            background: #f8f9fc;
+            border-radius: 10px;
+            padding: 20px;
+            box-shadow: inset 0 2px 4px rgba(0, 0, 0, 0.02);
+            border: 1px solid #e9ecef;
+        }
+        
+        /* Header gradient animation */
+        .header-gradient {
+            background: linear-gradient(135deg, #667eea 0%, #764ba2 50%, #667eea 100%);
+            background-size: 200% 200%;
+            animation: gradientShift 4s ease infinite;
+        }
+        
+        @keyframes gradientShift {
+            0% { background-position: 0% 50%; }
+            50% { background-position: 100% 50%; }
+            100% { background-position: 0% 50%; }
+        }
+        
+        /* Status indicator */
+        .status-success {
+            border-left: 4px solid #00b894 !important;
+            background-color: #d1f2eb !important;
         }
-        .api-key-box {
-            border: 2px solid #e1e5e9;
-            border-radius: 8px;
-            padding: 15px;
-            margin-bottom: 20px;
-            background-color: #f8f9fa;
+        
+        .status-error {
+            border-left: 4px solid #e17055 !important;
+            background-color: #fadbd8 !important;
+        }
+        
+        /* Responsive design */
+        @media (max-width: 768px) {
+            .gradio-container {
+                padding: 10px !important;
+            }
         }
         """
     ) as app:
         
         # Header
         gr.HTML("""
-        <div style="text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 10px; margin-bottom: 20px;">
-            <h1 style="color: white; margin: 0; font-size: 2.5em; font-weight: bold;">🚀 Nexus AI Assistant</h1>
-            <p style="color: white; margin: 10px 0 0 0; font-size: 1.2em;">Creative Multimodal AI Powered by AFM-4.5B</p>
+        <div class="header-gradient" style="text-align: center; padding: 30px; border-radius: 15px; margin-bottom: 25px;">
+            <h1 style="color: white; margin: 0; font-size: 3em; font-weight: 700; text-shadow: 2px 2px 4px rgba(0,0,0,0.3);">
+                🚀 Nexus AI Assistant
+            </h1>
+            <p style="color: white; margin: 15px 0 0 0; font-size: 1.3em; text-shadow: 1px 1px 2px rgba(0,0,0,0.2);">
+                Creative Multimodal AI Powered by AFM-4.5B
+            </p>
         </div>
         """)
         
         # API Key Section
-        with gr.Group(elem_classes=["api-key-box"]):
-            gr.HTML("<h3>🔑 API Configuration</h3>")
+        with gr.Group(elem_classes=["api-key-section"]):
+            gr.HTML("<h3 style='margin-top: 0; color: #2d3748;'>🔑 API Configuration</h3>")
             with gr.Row():
                 api_key_input = gr.Textbox(
                     label="Together AI API Key",
                     type="password",
                     placeholder="Enter your Together AI API key here...",
-                    scale=3
+                    scale=3,
+                    container=True
+                )
+                api_key_btn = gr.Button(
+                    "Initialize API Key", 
+                    variant="primary", 
+                    scale=1,
+                    elem_classes=["primary-button"]
                 )
-                api_key_btn = gr.Button("Initialize API Key", variant="primary", scale=1)
             
             api_key_status = gr.Textbox(
                 label="Status",
                 interactive=False,
-                value="Please enter your API key to get started"
+                value="Please enter your API key to get started",
+                elem_classes=["scrollable-textarea"]
             )
             
             # Hidden state to track API key status
@@ -379,23 +504,32 @@ def create_nexus_interface():
             
             # Text Chat Tab
             with gr.Tab("💬 Text Chat"):
-                with gr.Column():
-                    text_input = gr.Textbox(
-                        label="Your Message",
-                        placeholder="Ask me anything! I can help with creative tasks, analysis, problem-solving...",
-                        lines=3
-                    )
-                    text_btn = gr.Button("Send Message", variant="primary")
-                    text_output = gr.Textbox(
-                        label="Nexus AI Response",
-                        lines=8,
-                        interactive=False
-                    )
+                with gr.Row():
+                    with gr.Column(scale=1, elem_classes=["input-card"]):
+                        text_input = gr.Textbox(
+                            label="Your Message",
+                            placeholder="Ask me anything! I can help with creative tasks, analysis, problem-solving...",
+                            lines=4,
+                            elem_classes=["scrollable-textarea"]
+                        )
+                        text_btn = gr.Button(
+                            "Send Message", 
+                            variant="primary",
+                            elem_classes=["primary-button"]
+                        )
+                    
+                    with gr.Column(scale=1, elem_classes=["output-card"]):
+                        text_output = gr.Textbox(
+                            label="Nexus AI Response",
+                            lines=10,
+                            interactive=False,
+                            elem_classes=["scrollable-textarea"]
+                        )
             
             # PDF Analysis Tab
             with gr.Tab("📄 PDF Analysis"):
                 with gr.Row():
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["input-card"]):
                         pdf_file = gr.File(
                             label="Upload PDF",
                             file_types=[".pdf"]
@@ -403,21 +537,27 @@ def create_nexus_interface():
                         pdf_question = gr.Textbox(
                             label="Question about PDF (optional)",
                             placeholder="What would you like to know about this document?",
-                            lines=2
+                            lines=3,
+                            elem_classes=["scrollable-textarea"]
+                        )
+                        pdf_btn = gr.Button(
+                            "Analyze PDF", 
+                            variant="primary",
+                            elem_classes=["primary-button"]
                         )
-                        pdf_btn = gr.Button("Analyze PDF", variant="primary")
                     
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["output-card"]):
                         pdf_output = gr.Textbox(
                             label="Analysis Result",
                             lines=12,
-                            interactive=False
+                            interactive=False,
+                            elem_classes=["scrollable-textarea"]
                         )
             
             # Image Analysis Tab
             with gr.Tab("🖼️ Image Analysis"):
                 with gr.Row():
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["input-card"]):
                         image_file = gr.Image(
                             label="Upload Image",
                             type="filepath"
@@ -425,21 +565,27 @@ def create_nexus_interface():
                         image_question = gr.Textbox(
                             label="Question about Image (optional)",
                             placeholder="What would you like to know about this image?",
-                            lines=2
+                            lines=3,
+                            elem_classes=["scrollable-textarea"]
+                        )
+                        image_btn = gr.Button(
+                            "Analyze Image", 
+                            variant="primary",
+                            elem_classes=["primary-button"]
                         )
-                        image_btn = gr.Button("Analyze Image", variant="primary")
                     
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["output-card"]):
                         image_output = gr.Textbox(
                             label="Analysis Result",
                             lines=12,
-                            interactive=False
+                            interactive=False,
+                            elem_classes=["scrollable-textarea"]
                         )
             
             # Voice Processing Tab
             with gr.Tab("🎤 Voice Processing"):
                 with gr.Row():
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["input-card"]):
                         audio_file = gr.Audio(
                             label="Upload Audio",
                             type="filepath"
@@ -447,21 +593,27 @@ def create_nexus_interface():
                         audio_question = gr.Textbox(
                             label="Additional Question (optional)",
                             placeholder="Any specific question about the audio content?",
-                            lines=2
+                            lines=3,
+                            elem_classes=["scrollable-textarea"]
+                        )
+                        audio_btn = gr.Button(
+                            "Process Audio", 
+                            variant="primary",
+                            elem_classes=["primary-button"]
                         )
-                        audio_btn = gr.Button("Process Audio", variant="primary")
                     
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["output-card"]):
                         audio_output = gr.Textbox(
                             label="Processing Result",
                             lines=12,
-                            interactive=False
+                            interactive=False,
+                            elem_classes=["scrollable-textarea"]
                         )
             
             # Code Executor Tab
             with gr.Tab("⚡ Code Executor"):
                 with gr.Row():
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["input-card"]):
                         code_input = gr.Code(
                             label="Code Input",
                             language="python",
@@ -480,31 +632,57 @@ def create_nexus_interface():
                                 label="Action",
                                 scale=1
                             )
-                        code_btn = gr.Button("Process Code", variant="primary")
+                        code_btn = gr.Button(
+                            "Process Code", 
+                            variant="primary",
+                            elem_classes=["primary-button"]
+                        )
                     
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=1, elem_classes=["output-card"]):
                         code_output = gr.Textbox(
                             label="Result & Analysis",
                             lines=15,
-                            interactive=False
+                            interactive=False,
+                            elem_classes=["scrollable-textarea"]
                         )
             
             # Memory & History Tab
             with gr.Tab("🧠 Memory & History"):
-                with gr.Column():
-                    gr.HTML("<h3>Conversation Memory</h3>")
+                with gr.Column(elem_classes=["input-card"]):
+                    gr.HTML("<h3 style='margin-top: 0;'>Conversation Memory</h3>")
                     gr.HTML("<p>Nexus AI remembers your interactions and can connect insights across different input types.</p>")
                     
-                    history_btn = gr.Button("Show Recent History", variant="secondary")
+                    with gr.Row():
+                        history_btn = gr.Button(
+                            "Show Recent History", 
+                            variant="secondary",
+                            elem_classes=["secondary-button"],
+                            scale=1
+                        )
+                        clear_btn = gr.Button(
+                            "Clear History", 
+                            variant="secondary",
+                            elem_classes=["danger-button"],
+                            scale=1
+                        )
+                    
                     history_output = gr.Textbox(
                         label="Conversation History",
                         lines=15,
-                        interactive=False
+                        interactive=False,
+                        elem_classes=["scrollable-textarea"]
                     )
         
         # Event handlers
+        def update_api_status(api_key):
+            message, status = initialize_api_key(api_key)
+            if status == "success":
+                return gr.update(value=message, elem_classes=["scrollable-textarea", "status-success"]), status
+            else:
+                return gr.update(value=message, elem_classes=["scrollable-textarea", "status-error"]), status
+        
         api_key_btn.click(
-            fn=initialize_api_key,
+            fn=update_api_status,
             inputs=[api_key_input],
             outputs=[api_key_status, api_key_state]
         )
@@ -544,11 +722,20 @@ def create_nexus_interface():
             outputs=[history_output]
         )
         
+        clear_btn.click(
+            fn=clear_conversation_history,
+            outputs=[history_output]
+        )
+        
         # Footer
         gr.HTML("""
-        <div style="text-align: center; padding: 20px; margin-top: 30px; border-top: 1px solid #e1e5e9;">
-            <p style="color: #666;">🚀 <strong>Nexus AI Assistant</strong> - Powered by AFM-4.5B | Built with ❤️ using Gradio</p>
-            <p style="color: #888; font-size: 0.9em;">Multi-modal AI assistant for creative and analytical tasks</p>
+        <div style="text-align: center; padding: 25px; margin-top: 30px; border-top: 2px solid #e9ecef; background: linear-gradient(135deg, #f8f9fa 0%, #e9ecef 100%); border-radius: 10px;">
+            <p style="color: #495057; font-size: 1.1em; margin: 0;">
+                🚀 <strong>Nexus AI Assistant</strong> - Powered by AFM-4.5B | Built with ❤️ using Gradio
+            </p>
+            <p style="color: #6c757d; font-size: 0.9em; margin: 5px 0 0 0;">
+                Multi-modal AI assistant for creative and analytical tasks
+            </p>
         </div>
         """)