Update app.py

app.py

@@ -30,7 +30,11 @@ class HuggingFaceInferenceWrapper:
         # Call inference API with raw_response=True to get the raw Response object
         response = self.inference_api(inputs=prompt, raw_response=True)
 
-        # Parse the JSON response
+        # Check if the response is a string
+        if isinstance(response, str):
+            return response.strip()
+
+        # Parse the JSON response if the response is not a string
         if hasattr(response, 'json'):
             json_response = response.json()
             # Extract the relevant information from the JSON response
@@ -42,7 +46,12 @@ class HuggingFaceInferenceWrapper:
                 return json_response.get('generated_text', '').strip()
 
         # Fallback: return the raw response text if JSON parsing fails
-        return response.text.strip()
+        if hasattr(response, 'text'):
+            return response.text.strip()
+
+        # If none of the above conditions are met, return an empty string or handle the error appropriately
+        return ""
+