Update app.py

app.py

@@ -1009,13 +1009,15 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                             outputs=[search_mode]
                         )
 
-
                         async def handle_generation(query, setting, is_search):
-                            if is_search:
-                                result = await execute_search_and_generate(query, setting)
-                        else:
-                            result = await demo_instance.generation_code(query, setting)
-                        return result
+                            try:
+                                if is_search:
+                                    return await execute_search_and_generate(query, setting)
+                                return await demo_instance.generation_code(query, setting)
+                            except Exception as e:
+                                print(f"Generation error: {str(e)}")
+                                return ["", None, gr.update(active_key="error"), gr.update(open=False)]
+
 
                         btn.click(
                             fn=handle_generation,