Create main.py

main.py

@@ -0,0 +1,65 @@
+import os
+import PyPDF2
+from fastapi import FastAPI, File, UploadFile, Form
+from fastapi.responses import JSONResponse
+from openai import OpenAI
+
+app = FastAPI()
+
+# ✅ Initialize OpenAI client (use environment variable for safety)
+client = OpenAI(api_key=os.getenv("OPENAI_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:
+            text += page_text + "\n"
+    return text
+
+
+@app.post("/generate-questions")
+async def generate_questions(query: str = Form(...), pdf: UploadFile = File(...)):
+    try:
+        # Read uploaded PDF
+        pdf_text = extract_pdf_text(pdf.file)
+
+        # 🔥 Call OpenAI GPT-4o-mini 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": "user",
+                    "content": f"""
+Here is the PDF content (may be truncated):
+{pdf_text[:15000]}
+
+Now, answer this query: {query}
+"""
+                }
+            ],
+            temperature=0.3,
+            max_tokens=1200
+        )
+
+        return JSONResponse({
+            "success": True,
+            "answer": chat_completion.choices[0].message.content
+        })
+
+    except Exception as e:
+        return JSONResponse({"success": False, "error": str(e)})
+
+
+@app.get("/")
+def home():
+    return {"message": "✅ PDF Query API is running!"}