Update app.py

app.py

@@ -307,12 +307,12 @@ class GradioInterface:
                 outputs=[analysis_evaluation, refined_prompt, explanation_of_refinements, full_response_json]
             )
 
-            # In the __init__ method, modify the click event:
+            # In the __init__ method of GradioInterface class:
             apply_button.click(
                 fn=self.apply_prompts,
                 inputs=[prompt_text, refined_prompt, apply_model],
                 outputs=[original_output, refined_output],
-                queue=True  # Enable queuing for streaming
+                api_name="apply_prompts"  # Optional: adds API endpoint
             )
 
     def refine_prompt(self, prompt: str, meta_prompt_choice: str) -> tuple:
@@ -329,21 +329,18 @@ class GradioInterface:
         )
 
     def apply_prompts(self, original_prompt: str, refined_prompt: str, model: str):
-        async def stream_generator():
-            original_output = ""
-            refined_output = ""
+        try:
+            # Process original prompt
+            original_output = self.prompt_refiner.apply_prompt(original_prompt, model)
+            # Process refined prompt
+            refined_output = self.prompt_refiner.apply_prompt(refined_prompt, model)
             
-            # Stream original prompt
-            for response in self.prompt_refiner.apply_prompt(original_prompt, model):
-                original_output = response
-                yield original_output, refined_output
-                
-            # Stream refined prompt
-            for response in self.prompt_refiner.apply_prompt(refined_prompt, model):
-                refined_output = response
-                yield original_output, refined_output
-        
-        return stream_generator()
+            # Return both outputs directly
+            return original_output, refined_output
+            
+        except Exception as e:
+            # Return error messages for both outputs in case of failure
+            return f"Error: {str(e)}", f"Error: {str(e)}"