Update app.py

app.py

@@ -27,15 +27,23 @@ class HuggingFaceInferenceWrapper:
         self.inference_api = inference_api
 
     def generate(self, prompt: str, **kwargs) -> str:
-        # Call inference API - returns string directly
-        response = self.inference_api(inputs=prompt)
-        if isinstance(response, str):
-            return response.strip()
-        elif hasattr(response, 'text'):
-            return response.text.strip()
-        else:
-            # Handle JSON response
-            return str(response)
+        # 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
+        if hasattr(response, 'json'):
+            json_response = response.json()
+            # Extract the relevant information from the JSON response
+            # This depends on the structure of the JSON response from the API
+            # For example, if the response contains a 'generated_text' field:
+            if isinstance(json_response, list) and len(json_response) > 0:
+                return json_response[0].get('generated_text', '').strip()
+            elif isinstance(json_response, dict):
+                return json_response.get('generated_text', '').strip()
+
+        # Fallback: return the raw response text if JSON parsing fails
+        return response.text.strip()
+