Update app.py

app.py

@@ -1,60 +1,47 @@
 import gradio as gr
 import anthropic
 import PyPDF2
-import io
 import os
 
-# Claude client
 client = anthropic.Anthropic(api_key=os.environ.get("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"))
 
-# Store uploaded book text in memory
 book_storage = {}
 
 def extract_pdf_text(pdf_file):
-    """Extract text from PDF with page numbers"""
     if pdf_file is None:
         return None, "No file uploaded"
-    
     try:
         reader = PyPDF2.PdfReader(pdf_file)
         pages = []
         for i, page in enumerate(reader.pages):
             text = page.extract_text()
-            if text.strip():
+            if text and text.strip():
                 pages.append(f"[PAGE {i+1}]\n{text}")
-        
         full_text = "\n\n".join(pages)
         return full_text, f"✅ {len(reader.pages)} pages extracted successfully!"
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
-
 def upload_book(book_name, pdf_file):
-    """Admin uploads a book PDF"""
     if pdf_file is None:
         return "Please select a PDF file."
-    
     text, msg = extract_pdf_text(pdf_file)
     if text:
-        book_storage[book_name] = {
-            "text": text,
-            "pages": text.count("[PAGE ")
-        }
+        book_storage[book_name] = {"text": text, "pages": text.count("[PAGE ")}
         return f"✅ '{book_name}' uploaded! {msg}"
     return msg
 
-
 def ask_question(book_name, question, chat_history):
-    """Ask question from uploaded book"""
     if not question.strip():
         return chat_history, ""
-    
+
     if book_name not in book_storage:
-        chat_history.append((question, f"⚠️ Please upload '{book_name}' PDF first (Admin panel)."))
+        chat_history.append({"role": "user", "content": question})
+        chat_history.append({"role": "assistant", "content": f"⚠️ Please upload '{book_name}' PDF first from the Admin Upload tab."})
         return chat_history, ""
-    
+
     book_text = book_storage[book_name]["text"]
-    
+
     system_prompt = f"""You are a sacred scripture scholar for the AI Sacred Library.
 
 You have the TEXT of: {book_name}
@@ -62,59 +49,52 @@ You have the TEXT of: {book_name}
 RULES:
 1. Search the provided book text FIRST
 2. Give EXACT original text from the book (in original language if present)
-3. Always mention PAGE NUMBER using [PAGE X] markers
+3. Always mention PAGE NUMBER using [PAGE X] markers in the text
 4. Provide English translation
 5. Provide Tafseer/Commentary/Summary
-6. Format response clearly with sections:
-   📖 ORIGINAL TEXT (Page X):
-   [exact text]
-   
-   🌐 TRANSLATION:
-   [translation]
-   
-   📝 TAFSEER/COMMENTARY:
-   [explanation]
-   
-   📌 SOURCE: {book_name}, Page X
+6. Format response clearly:
+
+📖 ORIGINAL TEXT (Page X):
+[exact text from book]
+
+🌐 TRANSLATION:
+[English translation]
+
+📝 TAFSEER / COMMENTARY:
+[explanation and context]
+
+📌 SOURCE: {book_name}, Page X
 
 BOOK CONTENT:
 {book_text[:60000]}"""
 
     messages = []
-    for human, assistant in chat_history[-5:]:
-        messages.append({"role": "user", "content": human})
-        messages.append({"role": "assistant", "content": assistant})
+    for msg in chat_history[-10:]:
+        messages.append({"role": msg["role"], "content": msg["content"]})
     messages.append({"role": "user", "content": question})
 
     response = client.messages.create(
-        model="claude-opus-4-5",
+        model="claude-sonnet-4-20250514",
         max_tokens=1500,
         system=system_prompt,
         messages=messages
     )
-    
+
     answer = response.content[0].text
-    chat_history.append((question, answer))
+    chat_history.append({"role": "user", "content": question})
+    chat_history.append({"role": "assistant", "content": answer})
     return chat_history, ""
 
-
-# ── UI ──
 BOOKS = [
     "Al-Quran", "Bhagavad Gita", "Holy Bible", "Torah",
     "Tripitaka (Buddhism)", "The Vedas", "Guru Granth Sahib", "Avesta"
 ]
 
-with gr.Blocks(
-    title="Sacred Library — AI Scripture Database",
-    theme=gr.themes.Base(
-        primary_hue="green",
-        neutral_hue="slate",
-    ),
-    css="""
-    .gradio-container { background: #0a0a0f !important; color: #e8e0d0 !important; }
-    h1 { color: #f0e8d8 !important; text-align: center; }
-    """
-) as demo:
+css = """
+.gradio-container { max-width: 900px !important; margin: auto !important; }
+"""
+
+with gr.Blocks(title="Sacred Library — AI Scripture Database") as demo:
 
     gr.Markdown("""
     # 📚 Sacred Library
@@ -124,42 +104,38 @@ with gr.Blocks(
     """)
 
     with gr.Tabs():
-        
-        # ── CHAT TAB ──
+
         with gr.Tab("💬 Chat with Scripture"):
-            with gr.Row():
-                book_select = gr.Dropdown(
-                    choices=BOOKS,
-                    value="Al-Quran",
-                    label="📖 Select Scripture",
-                    scale=1
-                )
-            
+            book_select = gr.Dropdown(
+                choices=BOOKS,
+                value="Al-Quran",
+                label="📖 Select Scripture"
+            )
+
             chatbot = gr.Chatbot(
                 label="Sacred Library AI",
                 height=450,
-                bubble_full_width=False,
+                type="messages"
             )
-            
+
             with gr.Row():
                 question_input = gr.Textbox(
-                    placeholder='Ask anything... e.g. "page 1", "what is karma", "purpose of life", "Surah Al-Fatiha"',
+                    placeholder='e.g. "page 1", "what is karma", "purpose of life", "Surah Al-Fatiha"',
                     label="Your Question",
                     scale=4
                 )
                 ask_btn = gr.Button("Ask →", variant="primary", scale=1)
-            
+
             gr.Examples(
                 examples=[
                     ["Al-Quran", "Show me page 1"],
                     ["Al-Quran", "What does Quran say about purpose of life?"],
                     ["Bhagavad Gita", "What is karma according to Gita?"],
                     ["Holy Bible", "Genesis chapter 1 verse 1"],
-                    ["Al-Quran", "What is Surah Al-Fatiha?"],
                 ],
                 inputs=[book_select, question_input]
             )
-            
+
             ask_btn.click(
                 ask_question,
                 inputs=[book_select, question_input, chatbot],
@@ -171,21 +147,18 @@ with gr.Blocks(
                 outputs=[chatbot, question_input]
             )
 
-        # ── ADMIN UPLOAD TAB ──
         with gr.Tab("🔒 Admin — Upload Books"):
             gr.Markdown("""
             ### Admin Book Upload Panel
             Upload authentic scripture PDFs here. Users will get exact quotes from your uploaded books.
-            
-            > ⚠️ **Note**: In production, add password protection using `gr.State` + login check.
-            > For now, restrict this tab by keeping the Space private on Hugging Face.
+            > Keep this Space **Private** on Hugging Face to restrict admin access.
             """)
-            
+
             with gr.Row():
                 upload_book_name = gr.Dropdown(
                     choices=BOOKS,
                     value="Al-Quran",
-                    label="Select Book to Upload",
+                    label="Select Book",
                     scale=1
                 )
                 upload_pdf = gr.File(
@@ -193,25 +166,16 @@ with gr.Blocks(
                     file_types=[".pdf"],
                     scale=2
                 )
-            
+
             upload_btn = gr.Button("📤 Upload & Index Book", variant="primary")
-            upload_status = gr.Textbox(label="Upload Status", interactive=False)
-            
+            upload_status = gr.Textbox(label="Status", interactive=False)
+
             upload_btn.click(
                 upload_book,
                 inputs=[upload_book_name, upload_pdf],
                 outputs=[upload_status]
             )
-            
-            gr.Markdown("""
-            ---
-            **Currently Indexed Books:**
-            Check the chat tab — if you can get answers from a book, it's indexed ✅
-            """)
 
-    gr.Markdown("""
-    ---
-    *Sacred Library — Built with Claude AI + Gradio + PyPDF2*
-    """)
+    gr.Markdown("---\n*Sacred Library — Built with Claude AI + Gradio + PyPDF2*")
 
-demo.launch()
+demo.launch()