Update app.py

app.py

@@ -626,30 +626,6 @@ class XylariaChat:
 
     def create_interface(self):
         def streaming_response(message, chat_history, image_filepath, math_ocr_image_path):
-            # Placeholder for image generation
-            loading_svg = """<svg width="256" height="256" viewBox="0 0 256 256" xmlns="http://www.w3.org/2000/svg">
-              <style>
-                rect {
-                  animation: fillAnimation 3s ease-in-out infinite;
-                }
-                @keyframes fillAnimation {
-                  0% { fill: #626262; }
-                  50% { fill: #111111; }
-                  100% { fill: #626262; }
-                }
-                text {
-                  font-family: 'Helvetica Neue', Arial, sans-serif;
-                  font-weight: 300;
-                  text-shadow: 0px 2px 4px rgba(0, 0, 0, 0.4);
-                }
-              </style>
-              <rect width="256" height="256" rx="20" fill="#888888" />
-              <text x="50%" y="50%" dominant-baseline="middle" text-anchor="middle" font-size="24" fill="white" opacity="0.8">
-                <tspan>{/}</tspan>
-                <tspan x="50%" dy="1.2em"></tspan>
-              </text>
-            </svg>"""
-
             if message.strip().lower().startswith("/image"):
                 
                 image_prompt = message.strip().lower()[len("/image"):].strip()
@@ -658,8 +634,7 @@ class XylariaChat:
 
                 
                 chat_history.append([message, ""])
-                chat_history.append(("", loading_svg))
-                yield "", chat_history, None, None, None
+                yield "", chat_history, None, None
 
                 
                 image_bytes = self.generate_image(image_prompt)
@@ -671,20 +646,18 @@ class XylariaChat:
                     
                     chat_history[-1] = ("", image_html)
 
-                   
                     self.conversation_history.append(ChatMessage(role="user", content=message).to_dict())
                     self.conversation_history.append(ChatMessage(role="assistant", content=image_html).to_dict())
 
-                    
                     self.save_chat()
                     all_chats = self.load_all_chats()
                     chat_titles = [f"{chat['timestamp']}: {chat['conversation'][0]['content'][:30]}..." if len(chat['conversation']) > 0 and chat['conversation'][0]['content'] else f"{chat['timestamp']}: Empty Chat" for chat in all_chats]
 
-                    yield "", chat_history, None, None, gr.update(choices=chat_titles, visible=True)
+                    yield "", chat_history, None, None
                 else:
                     
                     chat_history[-1] = ("", image_bytes)
-                    yield "", chat_history, None, None, None
+                    yield "", chat_history, None, None
                 return
 
             ocr_text = ""
@@ -692,7 +665,7 @@ class XylariaChat:
                 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, None
+                    yield "", updated_history, None, None
                     return
                 else:
                     message = f"Math OCR Result: {ocr_text}\n\nUser's message: {message}"
@@ -704,7 +677,7 @@ class XylariaChat:
                 
             if isinstance(response_stream, str):
                 updated_history = chat_history + [[message, response_stream]]
-                yield "", updated_history, None, None, None
+                yield "", updated_history, None, None
                 return
 
             full_response = ""
@@ -712,7 +685,7 @@ class XylariaChat:
             
             if isinstance(response_stream, str):
                 updated_history = chat_history + [[message, response_stream]]
-                yield "", updated_history, None, None, None
+                yield "", updated_history, None, None
                 return
 
             try:
@@ -722,12 +695,7 @@ class XylariaChat:
                         full_response += chunk.token.text
                         updated_history[-1][1] = full_response
                         
-                        
-                        
-                        #self.conversation_history.append(ChatMessage(role="user", content=message).to_dict())
-                        #self.conversation_history.append(ChatMessage(role="assistant", content=full_response).to_dict())
-                        
-                        yield "", updated_history, None, None, None
+                        yield "", updated_history, None, None
                         
             except Exception as e:
                 print(f"Streaming error: {e}")