Update app.py

app.py

@@ -2,7 +2,11 @@ import os
 from dotenv import load_dotenv
 import requests
 import gradio as gr
+import logging
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Load environment variables
 load_dotenv()
@@ -10,12 +14,23 @@ load_dotenv()
 # Access the Hugging Face API key
 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()
 
 def summarize(input_text):
@@ -28,10 +43,13 @@ def summarize(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