Update app.py

app.py

@@ -9,12 +9,18 @@ client = Client("HydroFlyer53/ThePickle", hf_token=os.environ["Key"])
 def chat_with_ai(message, history):
     """Function to get AI response from Hugging Face model with limited history tracking."""
 
-    # Retain only the last 3 exchanges in history
+    # Retain only the last 3 exchanges in history and ensure proper formatting
     trimmed_history = history[-3:] if history else []
-    formatted_history = "\n".join([f"User: {h[0]}\nAI: {h[1]}" for h in trimmed_history])
+    formatted_history = "\n".join([f"User: {h[0]}\nAI: {h[1]}" for h in trimmed_history if h])
+    
+    # Ensure the history is not empty or corrupted
+    if not formatted_history.strip():
+        formatted_history = "User: Hi\nAI: What's up?"
 
     # Combine history with new message
     prompt = f"{formatted_history}\nUser: {message}\nAI:"
+    
+    print("DEBUG: Prompt sent to AI:", prompt)  # Debugging line to check input
 
     result = client.predict(
         message=prompt,
@@ -30,7 +36,11 @@ def chat_with_ai(message, history):
         top_p=0.60,
         api_name="/chat"
     )
-
+    
+    # Prevent endless repetition
+    if result.strip().count("Sus AI:") > 3:
+        return "Yo, I think I glitched out—let’s try that again!"
+    
     return result
 
 # Gradio Chat Interface with history