Update app.py

app.py

@@ -61,4 +61,21 @@ def run_agent(user_input):
     
     print("Raw Model Response:", raw_response)  # Debug output (optional)
     
-    if not raw_respon
+    if not raw_response.strip():
+        return "I'm sorry, but I couldn't generate a response. Please try again."
+    
+    clean_response = remove_code_snippets(raw_response)
+    words = clean_response.split()
+    if len(set(words)) < 5:  # Check for repetition or overly brief output.
+        return "I'm unable to generate a meaningful response. Please refine your query."
+    return clean_response
+
+# Create a simple agent wrapper with a run method.
+class SimpleAgent:
+    def __init__(self, run_function):
+        self.run = run_function
+
+agent = SimpleAgent(run_agent)
+
+# Launch the Gradio UI using the simple agent.
+GradioUI(agent).launch()