Update app.py

app.py

@@ -27,6 +27,15 @@ def respond(
 
     response = ""
 
+    # Prevent infinite loops by limiting history and avoiding repeated responses
+    if len(history) > 5:  # Limit history to the last 5 exchanges
+        history = history[-5:]
+
+    # Detect if responses are getting repetitive and stop the loop
+    if len(set([h[1] for h in history])) == 1:  # All assistant's responses are the same
+        yield "It seems we're repeating ourselves. Let's move to a new topic."
+        return
+
     # Use text-generation instead of chat-completion
     for message in client.text_generation(
         prompt=prompt,
@@ -62,4 +71,4 @@ demo = gr.ChatInterface(
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()