uppdate app and requirement.txt

app.py

@@ -1,17 +1,10 @@
 import gradio as gr
-import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from langgraph.graph import StateGraph, END
-from typing import TypedDict, List, Dict, Optional
+from typing import TypedDict, List, Dict
 from datetime import datetime
-import json
 import re
-import numpy as np
-from sentence_transformers import SentenceTransformer
-import faiss
-import pickle
-import os
 
 # Enhanced State Management with RAG
 class MedicalState(TypedDict):
@@ -127,7 +120,7 @@ class EnhancedMedicalAssistant:
         self.conversation_count = {}
     
     def load_models(self):
-        """Load the AI models"""
+        """Load the AI models with fallback options"""
         print("Loading models...")
         try:
             # Llama-2 for conversation
@@ -152,16 +145,17 @@ class EnhancedMedicalAssistant:
                 device_map="auto"
             )
             print("Models loaded successfully!")
+            
         except Exception as e:
             print(f"Error loading models: {e}")
-            # Fallback - use only one model
+            # Fallback - use smaller models
             self.tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
             self.model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
             self.meditron_tokenizer = self.tokenizer
             self.meditron_model = self.model
     
     def setup_langgraph(self):
-        """Setup simplified LangGraph workflow"""
+        """Setup LangGraph workflow"""
         workflow = StateGraph(MedicalState)
         
         workflow.add_node("intake", self.patient_intake)
@@ -466,7 +460,7 @@ Be professional and caring.
 # Initialize the medical assistant
 medical_assistant = EnhancedMedicalAssistant()
 
-@spaces.GPU  
+# Gradio chat interface function
 def chat_interface(message, history):
     """Gradio chat interface"""
     try:
@@ -475,7 +469,7 @@ def chat_interface(message, history):
         print(f"Chat interface error: {e}")
         return f"I apologize, but I encountered an error. Please try rephrasing your question. Error: {str(e)}"
 
-# Create Gradio interface
+# Create Gradio interface with enhanced styling
 demo = gr.ChatInterface(
     fn=chat_interface,
     title="🏥 Medical AI Assistant with medRAG",
@@ -498,20 +492,39 @@ demo = gr.ChatInterface(
     ],
     theme="soft",
     css="""
+    /* Main container styling */
+    .gradio-container {
+        background: linear-gradient(to right, #f0f8ff, #f5f5f5);
+        font-family: 'Arial', sans-serif;
+    }
+    
+    /* Chat message styling */
     .message.user { 
         background-color: #e3f2fd; 
-        border-radius: 10px;
-        padding: 10px;
-        margin: 5px;
+        border-radius: 12px;
+        padding: 12px;
+        margin: 8px;
+        box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+        border-left: 4px solid #2196F3;
     }
+    
     .message.bot { 
         background-color: #f1f8e9; 
-        border-radius: 10px;
-        padding: 10px;
-        margin: 5px;
+        border-radius: 12px;
+        padding: 12px;
+        margin: 8px;
+        box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+        border-left: 4px solid #4CAF50;
+    }
+    
+    /* Enhanced medical styling */
+    .bot h2 {
+        color: #1976D2 !important;
+        border-bottom: 2px solid #E0E0E0 !important;
+        padding-bottom: 8px !important;
     }
     """
 )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)

requirements.txt

@@ -1,38 +1,45 @@
-# Core ML Libraries
+# Core dependencies
+gradio>=4.0.0
 torch>=2.0.0
 transformers>=4.30.0
-accelerate>=0.20.0
-bitsandbytes>=0.39.0
-
-# Gradio for UI
-gradio>=4.0.0
-spaces>=0.19.4
-
-# LangGraph for workflow management
-langgraph>=0.0.40
-langchain>=0.1.0
-langchain-core>=0.1.0
+langgraph>=0.3.27
+langchain-core>=0.2.38
 
-# RAG and Vector Search
+# Vector search and embeddings
 sentence-transformers>=2.2.2
 faiss-cpu>=1.7.4
-# For GPU support, use: faiss-gpu>=1.7.4
 
-# Data Processing
+# Data processing
 numpy>=1.24.0
-pandas>=2.0.0
-scipy>=1.10.0
-
-# Text Processing
-regex>=2023.0.0
-nltk>=3.8.0
-spacy>=3.6.0
-
-# Utility Libraries
-python-dateutil>=2.8.0
-requests>=2.31.0
-urllib3>=2.0.0
-
-# JSON and Data Handling
-jsonschema>=4.17.0
-pydantic>=2.0.0
+typing-extensions>=4.5.0
+
+# LangGraph ecosystem components
+langsmith>=0.1.63
+langgraph-sdk>=0.1.66
+langgraph-checkpoint>=2.0.23
+
+# Web serving utilities
+httpx>=0.25.0
+uvicorn>=0.26.0
+sse-starlette>=2.1.0,<2.2.0
+uvloop>=0.18.0
+httptools>=0.5.0
+
+# Serialization and utilities
+orjson>=3.9.7,<3.10.17
+jsonschema-rs>=0.20.0
+structlog>=24.1.0
+cloudpickle>=3.0.0
+tenacity>=8.0.0
+
+# Model acceleration (optional but recommended)
+accelerate>=0.20.0
+safetensors>=0.3.1
+bitsandbytes>=0.40.0
+
+# For Hugging Face model access
+huggingface_hub>=0.16.0
+
+# Optional - specific model support
+langchain_anthropic>=0.0.5
+langchain_openai>=0.0.2