Update app.py

app.py

@@ -21,7 +21,7 @@ class ChatMessage:
         return {"role": self.role, "content": self.content}
 
 class XylariaChat:
-    def _init_(self):
+    def __init__(self):
         self.hf_token = os.getenv("HF_TOKEN")
         if not self.hf_token:
             raise ValueError("HuggingFace token not found in environment variables")
@@ -34,8 +34,6 @@ class XylariaChat:
         self.image_api_url = "https://api-inference.huggingface.co/models/Salesforce/blip-image-captioning-large"
         self.image_api_headers = {"Authorization": f"Bearer {self.hf_token}"}
 
-        self.image_gen_client = InferenceClient("black-forest-labs/FLUX.1-schnell", token=self.hf_token)
-
         self.conversation_history = []
         self.persistent_memory = []
         self.memory_embeddings = None
@@ -93,8 +91,6 @@ class XylariaChat:
             "photosynthesis": "process used by plants to convert light to energy"
         }
 
-    # ... (other methods: update_internal_state, update_knowledge_graph, etc.) ...
-
     def update_internal_state(self, emotion_deltas, cognitive_load_deltas, introspection_delta, engagement_delta):
         for emotion, delta in emotion_deltas.items():
             if emotion in self.internal_state["emotions"]:
@@ -425,13 +421,6 @@ class XylariaChat:
 
         except Exception as e:
             return f"Error processing image: {str(e)}"
-    
-    def generate_image(self, prompt):
-        try:
-            image = self.image_gen_client.text_to_image(prompt)
-            return image
-        except Exception as e:
-            return f"Error generating image: {e}"
 
     def perform_math_ocr(self, image_path):
         try:
@@ -553,44 +542,12 @@ class XylariaChat:
     def create_interface(self):
         def streaming_response(message, chat_history, image_filepath, math_ocr_image_path):
             
-            if "/image" in message:
-                image_prompt = message.replace("/image", "").strip()
-
-                # Placeholder for image generation
-                placeholder_image = "data:image/svg+xml," + requests.utils.quote(f'''
-                    <svg width="256" height="256" viewBox="0 0 256 256" xmlns="http://www.w3.org/2000/svg">
-                        <rect width="256" height="256" rx="20" fill="#888888" />
-                        <animate attributeName="fill" values="#888888;#000000;#888888" dur="2s" repeatCount="indefinite" />
-                        <text x="50%" y="50%" dominant-baseline="middle" text-anchor="middle" font-size="48" fill="white">
-                            <tspan>/</tspan>
-                        </text>
-                    </svg>
-                ''')
-
-                updated_history = chat_history + [[message, gr.Image(value=placeholder_image, type="pil", visible=True)]]
-                yield "", updated_history, None, None, ""
-
-                try:
-                    generated_image = self.generate_image(image_prompt)
-                    
-                    updated_history[-1][1] = gr.Image(value=generated_image, type="pil", visible=True)
-                    yield "", updated_history, None, None, ""
-
-                    self.conversation_history.append(ChatMessage(role="user", content=message).to_dict())
-                    self.conversation_history.append(ChatMessage(role="assistant", content="Image generated").to_dict())
-                    
-                    return
-                except Exception as e:
-                    updated_history[-1][1] = f"Error generating image: {e}"
-                    yield "", updated_history, None, None, ""
-                    return
-            
             ocr_text = ""
             if math_ocr_image_path:
                 ocr_text = self.perform_math_ocr(math_ocr_image_path)
                 if ocr_text.startswith("Error"):
                     updated_history = chat_history + [[message, ocr_text]]
-                    yield "", updated_history, None, None, ""
+                    yield "", updated_history, None, None
                     return
                 else:
                     message = f"Math OCR Result: {ocr_text}\n\nUser's message: {message}"
@@ -602,7 +559,7 @@ class XylariaChat:
                 
             if isinstance(response_stream, str):
                 updated_history = chat_history + [[message, response_stream]]
-                yield "", updated_history, None, None, ""
+                yield "", updated_history, None, None
                 return
 
             full_response = ""
@@ -615,11 +572,11 @@ class XylariaChat:
                         full_response += chunk_content
                         
                         updated_history[-1][1] = full_response
-                        yield "", updated_history, None, None, ""
+                        yield "", updated_history, None, None
             except Exception as e:
                 print(f"Streaming error: {e}")
                 updated_history[-1][1] = f"Error during response: {e}"
-                yield "", updated_history, None, None, ""
+                yield "", updated_history, None, None
                 return
 
             full_response = self.adjust_response_based_on_state(full_response)
@@ -667,77 +624,41 @@ class XylariaChat:
                 self.conversation_history = self.conversation_history[-10:]
 
         custom_css = """
-        @import url('https://fonts.googleapis.com/css2?family=Source+Sans+Pro:wght@400;600;700&display=swap');
-
-        body {
-            background-color: #f5f5f5;
-            font-family: 'Source Sans Pro', sans-serif;
-        }
-
-        .gradio-container {
-            max-width: 900px;
-            margin: 0 auto;
-            border-radius: 10px;
-            box-shadow: 0px 4px 20px rgba(0, 0, 0, 0.1);
+        @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;
         }
-
-        .chatbot-container {
-            background-color: #fff;
-            border-radius: 10px;
-            padding: 20px;
-        }
-
         .chatbot-container .message {
-            font-family: 'Source Sans Pro', sans-serif;
-            font-size: 16px;
-            line-height: 1.6;
+            font-family: 'Inter', sans-serif !important;
         }
-
         .gradio-container input,
         .gradio-container textarea,
         .gradio-container button {
-            font-family: 'Source Sans Pro', sans-serif;
-            font-size: 16px;
-            border-radius: 8px;
+            font-family: 'Inter', sans-serif !important;
         }
-
         .image-container {
             display: flex;
             gap: 10px;
-            margin-bottom: 20px;
-            justify-content: center;
+            margin-bottom: 10px;
         }
-
         .image-upload {
-            border: 2px dashed #d3d3d3;
+            border: 1px solid #ccc;
             border-radius: 8px;
-            padding: 20px;
-            background-color: #fafafa;
-            text-align: center;
-            transition: all 0.3s ease;
-        }
-
-        .image-upload:hover {
-            background-color: #f0f0f0;
-            border-color: #b3b3b3;
+            padding: 10px;
+            background-color: #f8f8f8;
         }
-
         .image-preview {
-            max-width: 150px;
-            max-height: 150px;
+            max-width: 200px;
+            max-height: 200px;
             border-radius: 8px;
-            box-shadow: 0px 2px 5px rgba(0, 0, 0, 0.1);
         }
-
         .clear-button {
             display: none;
         }
-
         .chatbot-container .message {
             opacity: 0;
             animation: fadeIn 0.5s ease-in-out forwards;
         }
-
         @keyframes fadeIn {
             from {
                 opacity: 0;
@@ -748,109 +669,42 @@ class XylariaChat:
                 transform: translateY(0);
             }
         }
-
         .gr-accordion-button {
             background-color: #f0f0f0 !important;
             border-radius: 8px !important;
-            padding: 15px !important;
+            padding: 10px !important;
             margin-bottom: 10px !important;
             transition: all 0.3s ease !important;
             cursor: pointer !important;
-            border: none !important;
-            box-shadow: 0px 2px 5px rgba(0, 0, 0, 0.05) !important;
         }
-
         .gr-accordion-button:hover {
             background-color: #e0e0e0 !important;
-            box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1) !important;
+            box-shadow: 0px 2px 4px rgba(0, 0, 0, 0.1) !important;
         }
-
         .gr-accordion-active .gr-accordion-button {
             background-color: #d0d0d0 !important;
-            box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1) !important;
+            box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1) !important;
         }
-
         .gr-accordion-content {
             transition: max-height 0.3s ease-in-out !important;
             overflow: hidden !important;
             max-height: 0 !important;
         }
-
         .gr-accordion-active .gr-accordion-content {
             max-height: 500px !important;
         }
-
         .gr-accordion {
             display: flex;
             flex-direction: column-reverse;
         }
-
-        .chatbot-icon {
-            width: 40px;
-            height: 40px;
-            border-radius: 50%;
-            margin-right: 10px;
-        }
-
-        .user-message .message-row {
-            background-color: #e8f0fe;
-            border-radius: 10px;
-            padding: 10px;
-            margin-bottom: 10px;
-            border-top-right-radius: 2px;
-        }
-
-        .assistant-message .message-row {
-            background-color: #f0f0f0;
-            border-radius: 10px;
-            padding: 10px;
-            margin-bottom: 10px;
-            border-top-left-radius: 2px;
-        }
-
-        .user-message .message-icon {
-            background: url('https://img.icons8.com/color/48/000000/user.png') no-repeat center center;
-            background-size: contain;
-            width: 30px;
-            height: 30px;
-            margin-right: 10px;
-        }
-
-        .assistant-message .message-icon {
-            background: url('https://i.ibb.co/7b7hLGH/Senoa-Icon-1.png') no-repeat center center;
-            background-size: cover;
-            width: 40px;
-            height: 40px;
-            margin-right: 10px;
-            border-radius: 50%;
-        }
-
-        .message-text {
-            flex-grow: 1;
-        }
-
-        .message-row {
-            display: flex;
-            align-items: center;
-        }
         """
 
-        with gr.Blocks(theme=gr.themes.Soft(
-            primary_hue="slate",
-            secondary_hue="gray",
-            neutral_hue="gray",
-            font=["Source Sans Pro", "Arial", "sans-serif"],
-        ), css=custom_css) as demo:
+        with gr.Blocks(theme='soft', css=custom_css) as demo:
             with gr.Column():
                 chatbot = gr.Chatbot(
                     label="Xylaria 1.5 Senoa",
-                    height=600,
+                    height=500,
                     show_copy_button=True,
-                    elem_classes="chatbot-container",
-                    avatar_images=(
-                        "https://img.icons8.com/color/48/000000/user.png",  # User avatar
-                        "https://i.ibb.co/7b7hLGH/Senoa-Icon-1.png"  # Bot avatar
-                    )
                 )
 
                 with gr.Accordion("Image Input", open=False, elem_classes="gr-accordion"):
@@ -880,18 +734,18 @@ class XylariaChat:
                     btn = gr.Button("Send", scale=1)
 
                 with gr.Row():
-                    clear = gr.Button("Clear Conversation", variant="stop")
+                    clear = gr.Button("Clear Conversation")
                     clear_memory = gr.Button("Clear Memory")
 
                 btn.click(
                     fn=streaming_response,
                     inputs=[txt, chatbot, img, math_ocr_img],
-                    outputs=[txt, chatbot, img, math_ocr_img, txt]
+                    outputs=[txt, chatbot, img, math_ocr_img]
                 )
                 txt.submit(
                     fn=streaming_response,
                     inputs=[txt, chatbot, img, math_ocr_img],
-                    outputs=[txt, chatbot, img, math_ocr_img, txt]
+                    outputs=[txt, chatbot, img, math_ocr_img]
                 )
 
                 clear.click(