Update app.py

app.py

@@ -1,28 +1,39 @@
 import gradio as gr
-import tempfile
-from utils import extract_pdf_by_page, split_by_chapter, build_faiss_index, retrieve_text, generate_notes_questions
+import os
+
+from utils import (
+    extract_pdf_by_page, split_by_chapter,
+    build_faiss_index, retrieve_text, generate_notes_questions
+)
 
 chunks = []
 faiss_index = None
 chunk_texts = []
 groq_api_key = ""
 
-def upload_pdf(file, use_chapter_split):
-    global chunks, faiss_index, chunk_texts
-    with tempfile.NamedTemporaryFile(delete=False) as tmp:
-        tmp.write(file.read())
-        tmp_path = tmp.name
+#def upload_pdf(file, use_chapter_split):
+#    global chunks, faiss_index, chunk_texts
+#    with tempfile.NamedTemporaryFile(delete=False) as tmp:
+#        tmp.write(file.read())
+#        tmp_path = tmp.name
+
+#    pages = extract_pdf_by_page(tmp_path)
+#    chunks = split_by_chapter(pages) if use_chapter_split else pages
+#    faiss_index, chunk_texts, chunks = build_faiss_index(chunks)
+#    return f"✅ Uploaded and indexed {len(chunks)} chunks."
 
-    pages = extract_pdf_by_page(tmp_path)
+def upload_pdf(file_path, use_chapter_split):
+    global chunks, faiss_index, chunk_texts
+    pages = extract_pdf_by_page(file_path)
     chunks = split_by_chapter(pages) if use_chapter_split else pages
     faiss_index, chunk_texts, chunks = build_faiss_index(chunks)
     return f"✅ Uploaded and indexed {len(chunks)} chunks."
+    
 
-
-def extract_text_from_pdf(file_path):
-    doc = fitz.open(file_path)  # file_path is already a string
-    pages = [page.get_text() for page in doc]
-    return pages
+#def extract_text_from_pdf(file_path):
+#    doc = fitz.open(file_path)  # file_path is already a string
+#    pages = [page.get_text() for page in doc]
+#    return pages
 
     
 def set_api_key(api_key):
@@ -33,6 +44,9 @@ def set_api_key(api_key):
 def process_query(query):
     if not groq_api_key:
         return "❌ Please provide a valid Groq API key."
+    if not faiss_index:
+        return "❌ Please upload and process a PDF first."
+        
     results = retrieve_text(query, faiss_index, chunk_texts, chunks)
     output = generate_notes_questions(results[0]['text'], groq_api_key)
     return f"📘 **{results[0].get('title', 'Page')}** (pages: {results[0].get('pages')})\n\n{output}"