Update main.py

main.py

@@ -6,7 +6,7 @@ from openai import OpenAI
 
 app = FastAPI()
 
-# ✅ Initialize OpenAI client (use environment variable for safety)
+# ✅ Initialize OpenAI client
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 # 📖 Extract text from PDF
@@ -21,39 +21,39 @@ def extract_pdf_text(file) -> str:
 
 
 @app.post("/Query-pdf")
-async def generate_questions(query: str = Form(...), pdf: UploadFile = File(...)):
+async def query_pdf(query: str = Form(...), pdf: UploadFile = File(...)):
     try:
-        # Read uploaded PDF
+        # Extract text from PDF
         pdf_text = extract_pdf_text(pdf.file)
 
-        # 🔥 Call OpenAI GPT-4o-mini model
+        # Restrictive system instruction
+        system_prompt = (
+            "You are a strict PDF-based question-answering assistant. "
+            "You must answer ONLY using the text from the provided PDF. "
+            "If the answer cannot be found explicitly or clearly inferred "
+            "from the PDF, respond strictly with: 'The answer is not available in the PDF.' "
+            "Do not use outside knowledge or assumptions."
+        )
+
+        # Call OpenAI GPT model
         chat_completion = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
-                {
-                    "role": "system",
-                    "content": (
-                        "You are a helpful assistant that answers questions based strictly "
-                        "on the provided PDF content. If the answer cannot be found, say so."
-                    )
-                },
+                {"role": "system", "content": system_prompt},
                 {
                     "role": "user",
-                    "content": f"""
-Here is the PDF content (may be truncated):
-{pdf_text[:15000]}
-
-Now, answer this query: {query}
-"""
+                    "content": f"PDF Content:\n{pdf_text[:15000]}\n\nQuery: {query}"
                 }
             ],
-            temperature=0.3,
-            max_tokens=1200
+            temperature=0.0,  # ⚡ ensures deterministic factual answers
+            max_tokens=1000
         )
 
+        answer = chat_completion.choices[0].message.content.strip()
+
         return JSONResponse({
             "success": True,
-            "answer": chat_completion.choices[0].message.content
+            "answer": answer
         })
 
     except Exception as e:
@@ -62,4 +62,4 @@ Now, answer this query: {query}
 
 @app.get("/")
 def home():
-    return {"message": "✅ PDF Query API is running!"}
+    return {"message": "✅ Strict PDF Query API is running!"}