Update main.py

main.py

@@ -7,57 +7,56 @@ from openai import OpenAI
 app = FastAPI()
 
 # ✅ Initialize OpenAI client
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+api_key = os.getenv("OPENAI_API_KEY")
+if not api_key:
+    raise ValueError("❌ OPENAI_API_KEY not found. Please add it in Hugging Face Space secrets.")
+client = OpenAI(api_key=api_key)
 
 # 📖 Extract text from PDF
 def extract_pdf_text(file) -> str:
-    text = ""
-    reader = PyPDF2.PdfReader(file)
-    for page in reader.pages:
-        page_text = page.extract_text()
-        if page_text:
+    try:
+        reader = PyPDF2.PdfReader(file)
+        text = ""
+        for page in reader.pages:
+            page_text = page.extract_text() or ""
             text += page_text + "\n"
-    return text
+        return text
+    except Exception as e:
+        raise Exception(f"PDF extraction failed: {e}")
 
 @app.post("/Query-pdf")
 async def query_pdf(query: str = Form(...), pdf: UploadFile = File(...)):
     try:
-        # Extract text from PDF
         pdf_text = extract_pdf_text(pdf.file)
-
-        # Restrictive system instruction
         system_prompt = (
             "You are a polite and helpful assistant who answers questions strictly using the provided PDF text. "
-            "You may summarize or rephrase information as long as it is clearly derived from the PDF. "
-            "If the requested information cannot be found or confidently inferred from the PDF, "
-            "respond kindly and respectfully with: "
+            "If the information is not in the PDF, respond with: "
             "'I'm sorry, but the answer is not available in the PDF.'"
         )
 
-        # Call OpenAI GPT model
         chat_completion = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": system_prompt},
-                {
-                    "role": "user",
-                    "content": f"PDF Content:\n{pdf_text[:15000]}\n\nQuery: {query}"
-                }
+                {"role": "user", "content": f"PDF Content:\n{pdf_text[:15000]}\n\nQuery: {query}"}
             ],
             temperature=0.0,
             max_tokens=1000
         )
 
         answer = chat_completion.choices[0].message.content.strip()
-
-        return JSONResponse({
-            "success": True,
-            "answer": answer
-        })
+        return JSONResponse({"success": True, "answer": answer})
 
     except Exception as e:
+        import traceback
+        print("❌ ERROR:", traceback.format_exc())
         return JSONResponse({"success": False, "error": str(e)})
 
 @app.get("/")
 def home():
     return {"message": "✅ Strict PDF Query API is running!"}
+
+# ✅ Required for Hugging Face Space
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)