Update app.py

app.py

@@ -367,15 +367,22 @@ class GradioInterface:
             )
 
     def refine_prompt(self, prompt: str, meta_prompt_choice: str) -> tuple:
+    try:
         input_data = PromptInput(text=prompt, meta_prompt_choice=meta_prompt_choice)
-        # Get the RefinementOutput object
         result = self.prompt_refiner.refine_prompt(input_data)
         
-        # Now properly extract values from the RefinementOutput object
-        analysis_evaluation = f"\n\n{result.initial_prompt_evaluation}" if result.initial_prompt_evaluation else ""
-        refined_prompt = result.refined_prompt if result.refined_prompt else ""
-        explanation_refinements = result.explanation_of_refinements if result.explanation_of_refinements else ""
-        full_response = result.raw_content if result.raw_content else {}
+        # Ensure all values are strings or None
+        analysis_evaluation = str(result.initial_prompt_evaluation) if result.initial_prompt_evaluation else ""
+        refined_prompt = str(result.refined_prompt) if result.refined_prompt else ""
+        explanation_refinements = str(result.explanation_of_refinements) if result.explanation_of_refinements else ""
+        
+        # Create a safe JSON-serializable dictionary
+        full_response = {
+            "initial_prompt_evaluation": analysis_evaluation,
+            "refined_prompt": refined_prompt,
+            "explanation_of_refinements": explanation_refinements,
+            "raw_content": str(result.raw_content) if result.raw_content else ""
+        }
         
         return (
             analysis_evaluation,
@@ -383,6 +390,16 @@ class GradioInterface:
             explanation_refinements,
             full_response
         )
+    except Exception as e:
+        # Return safe default values in case of any error
+        error_response = {
+            "error": str(e),
+            "initial_prompt_evaluation": "",
+            "refined_prompt": "",
+            "explanation_of_refinements": "",
+            "raw_content": ""
+        }
+        return "", "", "", error_response
 
     def apply_prompts(self, original_prompt: str, refined_prompt: str, model: str):
         original_output = self.prompt_refiner.apply_prompt(original_prompt, model)