Create app.py

app.py

@@ -0,0 +1,54 @@
+import gradio as gr
+import tempfile
+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
+
+    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."
+
+def set_api_key(api_key):
+    global groq_api_key
+    groq_api_key = api_key
+    return "✅ API key set!"
+
+def process_query(query):
+    if not groq_api_key:
+        return "❌ Please provide a valid Groq API key."
+    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}"
+
+with gr.Blocks(title="RAG PDF Notes & Questions Generator") as demo:
+    gr.Markdown("# 📚 PDF Chapter Notes & Question Generator using Groq LLM")
+
+    with gr.Row():
+        pdf = gr.File(label="Upload PDF", file_types=[".pdf"])
+        use_chapter = gr.Checkbox(label="Split by Chapter", value=True)
+        upload_btn = gr.Button("Process PDF")
+
+    output_text = gr.Textbox(label="Upload Status")
+    upload_btn.click(fn=upload_pdf, inputs=[pdf, use_chapter], outputs=output_text)
+
+    api_key_input = gr.Textbox(label="🔐 Groq API Key", type="password")
+    api_btn = gr.Button("Set API Key")
+    api_output = gr.Textbox(label="API Key Status")
+    api_btn.click(fn=set_api_key, inputs=[api_key_input], outputs=api_output)
+
+    query_input = gr.Textbox(label="Enter chapter title or page keyword")
+    query_btn = gr.Button("Generate Notes & Questions")
+    result_output = gr.Markdown()
+    query_btn.click(fn=process_query, inputs=[query_input], outputs=[result_output])
+
+demo.launch()