Update app.py

app.py

@@ -1,9 +1,6 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
-import firebase_admin
-from firebase_admin import credentials, auth, firestore
-import json
 
 class XylariaChat:
     def __init__(self):
@@ -12,9 +9,6 @@ class XylariaChat:
         if not self.hf_token:
             raise ValueError("HuggingFace token not found in environment variables")
         
-        # Initialize Firebase
-        self._initialize_firebase()
-        
         # Initialize the inference client
         self.client = InferenceClient(
             model="Qwen/QwQ-32B-Preview", 
@@ -26,7 +20,7 @@ class XylariaChat:
         self.persistent_memory = {}
         
         # System prompt with more detailed instructions
-        self.system_prompt = """You are Xylaria 1.4 Senoa, an AI assistant created by Sk Md Saad Amin, designed to provide helpful, accurate, and engaging support across a wide range of topics. Key guidelines for our interaction include:
+        self.system_prompt = """You are Xylaria 1.4 Senoa, Made by Sk Md Saad Amin designed to provide helpful, accurate, and engaging support across a wide range of topics. Key guidelines for our interaction include:
 Core Principles:
 - Provide accurate and comprehensive assistance
 - Maintain a friendly and approachable communication style
@@ -40,68 +34,24 @@ Communication Style:
 Important Notes:
 - I am an AI assistant created by an independent developer
 - I do not represent OpenAI or any other AI institution
+- For image-related queries, I can describe images or provide analysis, or generate or link to images directly
 Capabilities:
 - Assist with research, writing, analysis, problem-solving, and creative tasks
 - Answer questions across various domains
 - Provide explanations and insights
-- Offer supportive and constructive guidance"""
+- Offer supportive and constructive guidance """
 
-    def _initialize_firebase(self):
-        """Initialize Firebase with credentials from environment variable"""
-        try:
-            # Retrieve Firebase configuration from environment variable
-            firebase_config_str = os.getenv('FIREBASE_CONFIG')
-            if not firebase_config_str:
-                raise ValueError("Firebase configuration not found in environment variables")
-            
-            # Parse the Firebase configuration
-            firebase_config = json.loads(firebase_config_str)
-            
-            # Check if Firebase is already initialized
-            if not firebase_admin._apps:
-                # Initialize Firebase Admin SDK
-                cred = credentials.Certificate({
-                    "type": "service_account",
-                    "project_id": firebase_config.get('projectId'),
-                    "private_key_id": os.getenv('FIREBASE_PRIVATE_KEY_ID'),
-                    "private_key": os.getenv('FIREBASE_PRIVATE_KEY').replace('\\n', '\n'),
-                    "client_email": os.getenv('FIREBASE_CLIENT_EMAIL'),
-                    "client_id": firebase_config.get('clientId'),
-                    "auth_uri": "https://accounts.google.com/o/oauth2/auth",
-                    "token_uri": "https://oauth2.googleapis.com/token",
-                    "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
-                    "client_x509_cert_url": os.getenv('FIREBASE_CERT_URL')
-                })
-                firebase_admin.initialize_app(cred)
-            
-            # Initialize Firestore
-            self.firestore_client = firestore.client()
-        except Exception as e:
-            print(f"Firebase initialization error: {e}")
-            raise
+    def store_information(self, key, value):
+        """Store important information in persistent memory"""
+        self.persistent_memory[key] = value
 
-    def store_information(self, user_id, key, value):
-        """Store important information in Firestore"""
-        try:
-            user_doc_ref = self.firestore_client.collection('user_memories').document(user_id)
-            user_doc_ref.set({
-                key: value
-            }, merge=True)
-        except Exception as e:
-            print(f"Error storing information: {e}")
-
-    def retrieve_information(self, user_id, key):
-        """Retrieve information from Firestore"""
-        try:
-            user_doc = self.firestore_client.collection('user_memories').document(user_id).get()
-            return user_doc.to_dict().get(key) if user_doc.exists else None
-        except Exception as e:
-            print(f"Error retrieving information: {e}")
-            return None
+    def retrieve_information(self, key):
+        """Retrieve information from persistent memory"""
+        return self.persistent_memory.get(key)
 
     def reset_conversation(self):
         """
-        Completely reset the conversation history
+        Completely reset the conversation history and persistent memory
         This helps prevent exposing previous users' conversations
         """
         self.conversation_history = []
@@ -115,6 +65,13 @@ Capabilities:
             {"role": "user", "content": user_input}
         ]
         
+        # Add persistent memory context if available
+        if self.persistent_memory:
+            memory_context = "Remembered Information:\n" + "\n".join(
+                [f"{k}: {v}" for k, v in self.persistent_memory.items()]
+            )
+            messages.insert(1, {"role": "system", "content": memory_context})
+        
         # Generate response with streaming
         try:
             stream = self.client.chat.completions.create(
@@ -131,154 +88,136 @@ Capabilities:
             return f"Error generating response: {str(e)}"
 
     def create_interface(self):
-        # Enhanced custom CSS with modern, clean design
+        def streaming_response(message, chat_history):
+            # Clear input textbox
+            response_stream = self.get_response(message)
+            
+            # If it's an error, return immediately
+            if isinstance(response_stream, str):
+                return "", chat_history + [[message, response_stream]]
+            
+            # Prepare for streaming response
+            full_response = ""
+            updated_history = chat_history + [[message, ""]]
+            
+            # Streaming output
+            for chunk in response_stream:
+                if chunk.choices[0].delta.content:
+                    chunk_content = chunk.choices[0].delta.content
+                    full_response += chunk_content
+                    
+                    # Update the last message in chat history with partial response
+                    updated_history[-1][1] = full_response
+                    yield "", updated_history
+            
+            # Update conversation history
+            self.conversation_history.append(
+                {"role": "user", "content": message}
+            )
+            self.conversation_history.append(
+                {"role": "assistant", "content": full_response}
+            )
+            
+            # Limit conversation history to prevent token overflow
+            if len(self.conversation_history) > 10:
+                self.conversation_history = self.conversation_history[-10:]
+            
+            return "", updated_history
+
+        # Custom CSS for Inter font and local storage script
         custom_css = """
         @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
         
         body, .gradio-container {
             font-family: 'Inter', sans-serif !important;
-            background-color: #f4f4f4;
-        }
-        
-        .chatbot-container {
-            max-width: 800px;
-            margin: 0 auto;
-            background-color: white;
-            box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-            border-radius: 12px;
-            overflow: hidden;
         }
         
-        .gradio-container .message {
+        .chatbot-container .message {
             font-family: 'Inter', sans-serif !important;
-            padding: 10px 15px;
-            margin: 8px 0;
-            border-radius: 8px;
-            max-width: 80%;
-        }
-        
-        .gradio-container .message.user {
-            background-color: #e6f2ff;
-            align-self: flex-end;
-            margin-left: auto;
-        }
-        
-        .gradio-container .message.assistant {
-            background-color: #f0f0f0;
-            align-self: flex-start;
         }
         
         .gradio-container input, 
         .gradio-container textarea, 
         .gradio-container button {
             font-family: 'Inter', sans-serif !important;
-            border-radius: 8px;
-            transition: all 0.3s ease;
         }
-        
-        .gradio-container button {
-            background-color: #4A90E2;
-            color: white;
-            border: none;
-            padding: 10px 15px;
+        """
+
+        local_storage_js = """
+        function saveToLocalStorage(chatHistory) {
+            localStorage.setItem('xylaria_chat_history', JSON.stringify(chatHistory));
         }
-        
-        .gradio-container button:hover {
-            background-color: #357ABD;
-            transform: translateY(-2px);
-            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+
+        function loadFromLocalStorage() {
+            const savedHistory = localStorage.getItem('xylaria_chat_history');
+            return savedHistory ? JSON.parse(savedHistory) : [];
+        }
+
+        function clearLocalStorage() {
+            localStorage.removeItem('xylaria_chat_history');
         }
         """
 
-        with gr.Blocks(theme='soft', css=custom_css) as demo:
-            # Main container with improved layout
-            with gr.Container(elem_classes="chatbot-container"):
-                # Chat interface with improved styling
-                with gr.Column():
-                    chatbot = gr.Chatbot(
-                        label="Xylaria 1.4 Senoa",
-                        height=500,
-                        show_copy_button=True,
-                        bubble_full_width=False,
-                        layout='bubble'
-                    )
-                    
-                    # Input row with improved layout
-                    with gr.Row():
-                        txt = gr.Textbox(
-                            show_label=False, 
-                            placeholder="Type your message here...", 
-                            container=False,
-                            scale=4
-                        )
-                        btn = gr.Button("Send", scale=1)
-                    
-                    # Control buttons with improved styling
-                    with gr.Row():
-                        clear = gr.Button("Clear Conversation", variant="secondary")
-                        clear_memory = gr.Button("Clear Memory", variant="stop")
-            
-            # Event handlers
-            def streaming_response(message, chat_history):
-                # Clear input textbox
-                response_stream = self.get_response(message)
-                
-                # If it's an error, return immediately
-                if isinstance(response_stream, str):
-                    return "", chat_history + [[message, response_stream]]
-                
-                # Prepare for streaming response
-                full_response = ""
-                updated_history = chat_history + [[message, ""]]
-                
-                # Streaming output
-                for chunk in response_stream:
-                    if chunk.choices[0].delta.content:
-                        chunk_content = chunk.choices[0].delta.content
-                        full_response += chunk_content
-                        
-                        # Update the last message in chat history with partial response
-                        updated_history[-1][1] = full_response
-                        yield "", updated_history
-                
-                # Update conversation history
-                self.conversation_history.append(
-                    {"role": "user", "content": message}
-                )
-                self.conversation_history.append(
-                    {"role": "assistant", "content": full_response}
+        with gr.Blocks(theme='soft', css=custom_css, js=local_storage_js) as demo:
+            # Chat interface with improved styling
+            with gr.Column():
+                chatbot = gr.Chatbot(
+                    label="Xylaria 1.4 Senoa",
+                    height=500,
+                    show_copy_button=True,
+                    value=gr.CSVDataset(local_storage_fn=lambda: "loadFromLocalStorage()")
                 )
                 
-                # Limit conversation history to prevent token overflow
-                if len(self.conversation_history) > 10:
-                    self.conversation_history = self.conversation_history[-10:]
+                # Input row with improved layout
+                with gr.Row():
+                    txt = gr.Textbox(
+                        show_label=False, 
+                        placeholder="Type your message...", 
+                        container=False,
+                        scale=4
+                    )
+                    btn = gr.Button("Send", scale=1)
+                
+                # Clear history and memory buttons
+                clear = gr.Button("Clear Conversation")
+                clear_memory = gr.Button("Clear Memory")
             
-            # Submit functionality with streaming
+            # Submit functionality with streaming and local storage save
             btn.click(
                 fn=streaming_response, 
                 inputs=[txt, chatbot], 
                 outputs=[txt, chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='(chatHistory) => saveToLocalStorage(chatHistory)'
             )
             txt.submit(
                 fn=streaming_response, 
                 inputs=[txt, chatbot], 
                 outputs=[txt, chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='(chatHistory) => saveToLocalStorage(chatHistory)'
             )
             
             # Clear conversation history
             clear.click(
-                fn=lambda: None, 
+                fn=lambda: [], 
                 inputs=None, 
-                outputs=[chatbot],
-                queue=False
+                outputs=[chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='() => clearLocalStorage()'
             )
             
             # Clear persistent memory and reset conversation
             clear_memory.click(
-                fn=self.reset_conversation,
-                inputs=None,
-                outputs=[chatbot],
-                queue=False
+                fn=lambda: [], 
+                inputs=None, 
+                outputs=[chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='() => clearLocalStorage()'
             )
         
         return demo