Update app.py

app.py

@@ -3,7 +3,7 @@ import anthropic
 import PyPDF2
 import os
 
-client = anthropic.Anthropic(api_key=os.environ.get("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"))
+client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
 
 book_storage = {}
 
@@ -27,7 +27,7 @@ def upload_book(book_name, pdf_file):
         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}
         return f"✅ '{book_name}' uploaded! {msg}"
     return msg
 
@@ -36,32 +36,30 @@ def ask_question(book_name, question, chat_history):
         return chat_history, ""
 
     if book_name not in book_storage:
-        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."})
+        chat_history.append((question, 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}
+You have the full 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 in the text
+1. Search the provided book text FIRST before answering
+2. Give EXACT original text from the book (keep original language like Arabic, Sanskrit, Hebrew)
+3. Always mention PAGE NUMBER — use the [PAGE X] markers in the text
 4. Provide English translation
-5. Provide Tafseer/Commentary/Summary
-6. Format response clearly:
+5. Provide Tafseer / Commentary / Summary
+6. Format your response like this:
 
 📖 ORIGINAL TEXT (Page X):
-[exact text from book]
+[exact text]
 
 🌐 TRANSLATION:
 [English translation]
 
 📝 TAFSEER / COMMENTARY:
-[explanation and context]
+[explanation]
 
 📌 SOURCE: {book_name}, Page X
 
@@ -69,20 +67,25 @@ BOOK CONTENT:
 {book_text[:60000]}"""
 
     messages = []
-    for msg in chat_history[-10:]:
-        messages.append({"role": msg["role"], "content": msg["content"]})
+    for human, assistant in (chat_history or [])[-5:]:
+        if human:
+            messages.append({"role": "user", "content": human})
+        if assistant:
+            messages.append({"role": "assistant", "content": assistant})
     messages.append({"role": "user", "content": question})
 
-    response = client.messages.create(
-        model="claude-sonnet-4-20250514",
-        max_tokens=1500,
-        system=system_prompt,
-        messages=messages
-    )
+    try:
+        response = client.messages.create(
+            model="claude-sonnet-4-20250514",
+            max_tokens=1500,
+            system=system_prompt,
+            messages=messages
+        )
+        answer = response.content[0].text
+    except Exception as e:
+        answer = f"❌ API Error: {str(e)}"
 
-    answer = response.content[0].text
-    chat_history.append({"role": "user", "content": question})
-    chat_history.append({"role": "assistant", "content": answer})
+    chat_history.append((question, answer))
     return chat_history, ""
 
 BOOKS = [
@@ -90,22 +93,19 @@ BOOKS = [
     "Tripitaka (Buddhism)", "The Vedas", "Guru Granth Sahib", "Avesta"
 ]
 
-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
     ### AI-Powered World Scripture Database
-    *Ask anything from any religious scripture — get exact quotes with page numbers, translations, and Tafseer*
+    *Exact quotes · Page numbers · Translation · Tafseer*
     ---
     """)
 
     with gr.Tabs():
 
         with gr.Tab("💬 Chat with Scripture"):
+
             book_select = gr.Dropdown(
                 choices=BOOKS,
                 value="Al-Quran",
@@ -114,13 +114,12 @@ with gr.Blocks(title="Sacred Library — AI Scripture Database") as demo:
 
             chatbot = gr.Chatbot(
                 label="Sacred Library AI",
-                height=450,
-                type="messages"
+                height=450
             )
 
             with gr.Row():
                 question_input = gr.Textbox(
-                    placeholder='e.g. "page 1", "what is karma", "purpose of life", "Surah Al-Fatiha"',
+                    placeholder='e.g. "page 1", "what is karma", "Surah Al-Fatiha", "Genesis chapter 1"',
                     label="Your Question",
                     scale=4
                 )
@@ -150,8 +149,8 @@ with gr.Blocks(title="Sacred Library — AI Scripture Database") as demo:
         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.
-            > Keep this Space **Private** on Hugging Face to restrict admin access.
+            Upload authentic scripture PDFs. Users get exact quotes from your books.
+            > 💡 Keep this Space **Private** on Hugging Face to restrict access.
             """)
 
             with gr.Row():
@@ -176,6 +175,6 @@ with gr.Blocks(title="Sacred Library — AI Scripture Database") as demo:
                 outputs=[upload_status]
             )
 
-    gr.Markdown("---\n*Sacred Library — Built with Claude AI + Gradio + PyPDF2*")
+    gr.Markdown("---\n*Sacred Library — Claude AI + Gradio + PyPDF2*")
 
 demo.launch()