Update app.py

app.py

@@ -307,7 +307,7 @@ 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, modify the click event:
             apply_button.click(
                 fn=self.apply_prompts,
                 inputs=[prompt_text, refined_prompt, apply_model],
@@ -329,14 +329,22 @@ class GradioInterface:
         )
 
     def apply_prompts(self, original_prompt: str, refined_prompt: str, model: str):
-        def stream_generator():
-            for response in self.prompt_refiner.apply_prompt(original_prompt, model):
-                yield response, ""
+        async def stream_generator():
+            original_output = ""
+            refined_output = ""
             
+            # 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):
-                current_output = yield "", response
+                refined_output = response
+                yield original_output, refined_output
         
         return stream_generator()
+