Update app.py

app.py

@@ -3,6 +3,11 @@ from dotenv import load_dotenv
 import requests
 import json
 import gradio as gr
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 # Load environment variables
 load_dotenv()
@@ -24,25 +29,27 @@ def get_completion(inputs, parameters=None):
     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}")
+    logger.debug(f"Sending request to {ENDPOINT_URL}")
+    logger.debug(f"Request headers: {headers}")
+    logger.debug(f"Request data: {json.dumps(data, indent=2)}")
 
     try:
         response = requests.post(ENDPOINT_URL, headers=headers, json=data)
-        print(f"Response status code: {response.status_code}")
-        print(f"Response content: {response.text}")
+        logger.debug(f"Response status code: {response.status_code}")
+        logger.debug(f"Response headers: {dict(response.headers)}")
+        logger.debug(f"Response content: {response.text}")
         
         response.raise_for_status()
         return response.json()
     except requests.exceptions.RequestException as e:
-        print(f"Request failed: {e}")
+        logger.error(f"Request failed: {e}")
         if hasattr(e, 'response') and e.response is not None:
-            print(f"Error response content: {e.response.text}")
+            logger.error(f"Error response content: {e.response.text}")
         return {"error": f"Request failed: {str(e)}"}
 
 def summarize(input_text):
     try:
+        logger.info(f"Received input text: {input_text[:100]}...")  # Log first 100 chars of input
         output = get_completion(input_text)
         if isinstance(output, list) and len(output) > 0 and 'summary_text' in output[0]:
             return output[0]['summary_text']
@@ -51,6 +58,7 @@ def summarize(input_text):
         else:
             return f"Unexpected response format: {output}"
     except Exception as e:
+        logger.exception("An error occurred during summarization")
         return f"An error occurred: {str(e)}"
 
 demo = gr.Interface(fn=summarize, inputs="text", outputs="text")