Update app.py

app.py

@@ -29,7 +29,7 @@ class XylariaChat:
             raise ValueError("HuggingFace token not found in environment variables")
 
         self.client = InferenceClient(
-            model="Qwen/QwQ-32B-Preview",
+            model="Qwen/Qwen-32B-Preview",
             api_key=self.hf_token
         )
 
@@ -395,7 +395,7 @@ class XylariaChat:
 
         try:
             self.client = InferenceClient(
-                model="Qwen/QwQ-32B-Preview",
+                model="Qwen/Qwen-32B-Preview",
                 api_key=self.hf_token
             )
         except Exception as e:
@@ -524,7 +524,7 @@ class XylariaChat:
 
             stream = self.client.chat_completion(
                 messages=messages,
-                model="Qwen/QwQ-32B-Preview",
+                model="Qwen/Qwen-32B-Preview",
                 temperature=0.7,
                 max_tokens=max_new_tokens,
                 top_p=0.9,
@@ -614,40 +614,12 @@ class XylariaChat:
             raise ValueError("Invalid chat index")
 
     def create_interface(self):
-        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; /* Choose a good font */
-              font-weight: 300; /* Slightly lighter font weight */
-              text-shadow: 0px 2px 4px rgba(0, 0, 0, 0.4); /* Subtle shadow */
-            }
-          </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>creating your image</tspan>
-            <tspan x="50%" dy="1.2em">with xylaria iris</tspan>
-          </text>
-        </svg>"""
-            
         def streaming_response(message, chat_history, image_filepath, math_ocr_image_path):
-            if message.strip().lower() == "/image":
-                # Extract image generation prompt from the last user message before /image
-                image_prompt = ""
-                for i in range(len(chat_history) - 1, -1, -1):
-                    if chat_history[i][0] and chat_history[i][0].strip().lower() != "/image":
-                        image_prompt = chat_history[i][0]
-                        break
-
+            if message.strip().lower().startswith("/image"):
+                # Extract image generation prompt from the message
+                image_prompt = message.strip().lower()[len("/image"):].strip()
                 if not image_prompt:
-                    image_prompt = "A realistic image"
+                    image_prompt = "A realistic image"  # Default prompt
 
                 # Generate image based on the extracted prompt
                 image_bytes = self.generate_image(image_prompt)
@@ -655,9 +627,9 @@ class XylariaChat:
                 if isinstance(image_bytes, bytes):
                     base64_image = base64.b64encode(image_bytes).decode("utf-8")
                     image_html = f'<img src="data:image/png;base64,{base64_image}" alt="Generated Image" style="max-width: 100%; max-height: 400px;">'
-                    
+
                     # Append the /image command and generated image to chat history
-                    chat_history.append([message, ""]) 
+                    chat_history.append([message, ""])
                     chat_history.append(["", image_html])
 
                     # Update conversation history
@@ -668,13 +640,13 @@ class XylariaChat:
                     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)
                 else:
                     chat_history.append([message, image_bytes])
                     yield "", chat_history, None, None, None
                 return
-            
+
             ocr_text = ""
             if math_ocr_image_path:
                 ocr_text = self.perform_math_ocr(math_ocr_image_path)