Update app.py

app.py

@@ -1,33 +1,44 @@
 import os
-from dotenv import load_dotenv, find_dotenv
-import requests, json
-from transformers import pipeline
+from dotenv import load_dotenv
+import requests
+import json
 import gradio as gr
 
+# Load environment variables
 load_dotenv()
+
 # Access the Hugging Face API key and endpoint URL
 hf_api_key = os.getenv('HF_API_KEY')
 MODEL_NAME = "sshleifer/distilbart-cnn-12-6"
 ENDPOINT_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
-#Summarization endpoint
 def get_completion(inputs, parameters=None):
     headers = {
         "Authorization": f"Bearer {hf_api_key}",
         "Content-Type": "application/json"
     }
     data = {
-        "inputs": inputs
+        "inputs": inputs,
+        "options": {"wait_for_model": True}
     }
     if parameters is not None:
         data.update({"parameters": parameters})
     
+    print(f"Sending request to {ENDPOINT_URL}")
+    print(f"Request headers: {headers}")
+    print(f"Request data: {data}")
+
     try:
         response = requests.post(ENDPOINT_URL, headers=headers, json=data)
+        print(f"Response status code: {response.status_code}")
+        print(f"Response content: {response.text}")
+        
         response.raise_for_status()
         return response.json()
     except requests.exceptions.RequestException as e:
         print(f"Request failed: {e}")
+        if hasattr(e, 'response') and e.response is not None:
+            print(f"Error response content: {e.response.text}")
         return {"error": f"Request failed: {str(e)}"}
 
 def summarize(input_text):
@@ -35,6 +46,8 @@ def summarize(input_text):
         output = get_completion(input_text)
         if isinstance(output, list) and len(output) > 0 and 'summary_text' in output[0]:
             return output[0]['summary_text']
+        elif isinstance(output, dict) and 'error' in output:
+            return f"API Error: {output['error']}"
         else:
             return f"Unexpected response format: {output}"
     except Exception as e: