Update app.py

app.py

@@ -1,57 +1,45 @@
 import os
 from dotenv import load_dotenv
 import requests
+import json
 import gradio as gr
-import logging
-
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
 
 # Load environment variables
 load_dotenv()
 
-# Access the Hugging Face API key
+# Access the Hugging Face API key and endpoint URL
 hf_api_key = os.getenv('HF_API_KEY')
-
-# Check if the API key is available
-if not hf_api_key:
-    logger.error("HF_API_KEY is not set in the environment variables.")
-    raise ValueError("HF_API_KEY is missing. Please set it in your environment or .env file.")
-
-# Log a part of the API key for verification (be careful not to log the entire key)
-logger.info(f"Using API key starting with: {hf_api_key[:4]}{'*' * (len(hf_api_key) - 4)}")
-
-# Set up the API endpoint
-API_URL = "https://api-inference.huggingface.co/models/sshleifer/distilbart-cnn-12-6"
-headers = {"Authorization": f"Bearer {hf_api_key}"}
-
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    if response.status_code != 200:
-        logger.error(f"API request failed with status code {response.status_code}")
-        logger.error(f"Response content: {response.text}")
-    return response.json()
+MODEL_NAME = "sshleifer/distilbart-cnn-12-6"
+ENDPOINT_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
+
+def get_completion(inputs, parameters=None):
+    headers = {
+        "Authorization": f"Bearer {hf_api_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "inputs": inputs
+    }
+    if parameters is not None:
+        data.update({"parameters": parameters})
+    
+    try:
+        response = requests.post(ENDPOINT_URL, headers=headers, json=data)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        print(f"Request failed: {e}")
+        return {"error": f"Request failed: {str(e)}"}
 
 def summarize(input_text):
     try:
-        output = query({
-            "inputs": input_text,
-            "parameters": {"max_length": 130, "min_length": 30}
-        })
-        
+        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:
-            logger.error(f"API Error: {output['error']}")
-            return f"API Error: {output['error']}"
         else:
-            logger.error(f"Unexpected response format: {output}")
             return f"Unexpected response format: {output}"
     except Exception as e:
-        logger.exception("An error occurred during summarization")
         return f"An error occurred: {str(e)}"
 
-# Create and launch the Gradio interface
 demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
 demo.launch()