Update app.py

app.py

@@ -114,15 +114,31 @@ async def chat(request: ChatRequest):
             "Provide concise answers to the exact question asked and nothing more.\n"
         )
 
-        seen_messages = set()
+
+        seen_user_messages = set()
+        seen_model_messages = set()
+        
         for msg in history:
-            if msg.role == "user" and msg.text.strip() not in seen_messages:
+            if msg.role == "user":
                 cleaned_user_text = clean_input_text(msg.text.strip())
-                conversation += f"<|user|>\n{cleaned_user_text}\n"
-                seen_messages.add(cleaned_user_text)
+                if cleaned_user_text and cleaned_user_text not in seen_user_messages:
+                    conversation += f"<|user|>\n{cleaned_user_text}\n"
+                    seen_user_messages.add(cleaned_user_text)
+        
             elif msg.role == "model":
                 cleaned_model_text = clean_input_text(msg.text.strip())
-                conversation += f"<|assistant|>\n{cleaned_model_text}\n"
+                if cleaned_model_text and cleaned_model_text not in seen_model_messages:
+                    conversation += f"<|assistant|>\n{cleaned_model_text}\n"
+                    seen_model_messages.add(cleaned_model_text)
+        # seen_messages = set()
+        # for msg in history:
+        #     if msg.role == "user" and msg.text.strip() not in seen_messages:
+        #         cleaned_user_text = clean_input_text(msg.text.strip())
+        #         conversation += f"<|user|>\n{cleaned_user_text}\n"
+        #         seen_messages.add(cleaned_user_text)
+        #     elif msg.role == "model":
+        #         cleaned_model_text = clean_input_text(msg.text.strip())
+        #         conversation += f"<|assistant|>\n{cleaned_model_text}\n"
 
 
         #conversation += f"<|user|>\n{user_message.strip()}\n<|assistant|>"