Improve error handling and response parsing for HF API

main.py

@@ -86,20 +86,34 @@ Feedback: This code could be improved by"""
         }
     }
 
-    response = requests.post(
-        f"https://api-inference.huggingface.co/models/{HF_MODEL_NAME}",
-        headers=headers,
-        json=payload
-    )
-
-    if response.status_code != 200:
-        raise RuntimeError(f"Hugging Face API error: {response.status_code} {response.text}")
-
-    response_data = response.json()
-    if isinstance(response_data, list) and len(response_data) > 0:
-        response_text = response_data[0].get("generated_text", "").strip()
-    else:
-        response_text = "Unable to generate a meaningful review."
+    try:
+        response = requests.post(
+            f"https://api-inference.huggingface.co/models/{HF_MODEL_NAME}",
+            headers=headers,
+            json=payload,
+            timeout=30
+        )
+
+        if response.status_code != 200:
+            print(f"HF API Error: {response.status_code} - {response.text}")
+            return "Consider adding proper documentation and error handling."
+
+        response_data = response.json()
+        print(f"HF API Response: {response_data}")
+        
+        if isinstance(response_data, list) and len(response_data) > 0:
+            generated_text = response_data[0].get("generated_text", "")
+            # Extract only the new generated part (after our prompt)
+            if generated_text.startswith(prompt):
+                response_text = generated_text[len(prompt):].strip()
+            else:
+                response_text = generated_text.strip()
+        else:
+            response_text = "Unable to generate a meaningful review."
+
+    except Exception as e:
+        print(f"HF API Exception: {e}")
+        return "Consider adding proper documentation and error handling."
 
     # Clean up the response
     response_text = response_text.strip()