Update app.py

app.py

@@ -1,16 +1,20 @@
 import gradio as gr
 import requests
+import os  # Import the os module to access environment variables
 
 def chat_with_pdf(content):
- api_key = os.getenv('CHATPDFKEY')
+    # Use the CHATPDFKEY environment variable for the API key
+    api_key = os.getenv('CHATPDFKEY')
+    if not api_key:
+        return "API key (CHATPDFKEY) not set in environment variables."
     
-headers = {
+    headers = {
         "x-api-key": api_key,
         "Content-Type": "application/json",
     }
 
-data = {
-        #"stream": True,
+    data = {
+        "stream": True,
         "sourceId": "src_REdjHAwIpm8Lhuk2HAtnm",  # Replace with your source ID
         "messages": [
             {
@@ -20,7 +24,7 @@ data = {
         ],
     }
 
-url = "https://api.chatpdf.com/v1/chats/message"
+    url = "https://api.chatpdf.com/v1/chats/message"
 
     try:
         response = requests.post(url, json=data, headers=headers, stream=True)
@@ -35,8 +39,10 @@ url = "https://api.chatpdf.com/v1/chats/message"
             return chat_response
         else:
             return "No data received"
+    except requests.exceptions.HTTPError as error:
+        return f"HTTP Error: {error.response.status_code} - {error.response.text}"
     except requests.exceptions.RequestException as error:
-        return str(error)
+        return f"Request Exception: {error}"
 
 # Create Gradio interface
 iface = gr.Interface(
@@ -44,7 +50,7 @@ iface = gr.Interface(
     inputs=gr.inputs.Textbox(lines=2, placeholder="Ask something..."),
     outputs="text",
     title="ChatPDF Query Interface",
-    description="Type your question to get answers from PDFs."
+    description="Type your question to get answers from PDFs. Make sure the CHATPDFKEY environment variable is set."
 )
 
 # Run the Gradio app