Update app.py

app.py

@@ -1,7 +1,5 @@
 import os
 import gradio as gr
-
-# ✅ LangChain imports
 from langchain.text_splitter import CharacterTextSplitter
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
@@ -9,25 +7,24 @@ from langchain.chains import ConversationalRetrievalChain
 from langchain_community.llms import HuggingFaceHub
 from langchain_community.document_loaders import PyPDFLoader
 
-# --- 1️⃣ Load your PDF ---
-current_dir = os.path.dirname(__file__)
-pdf_path = os.path.join(current_dir, "chimera.pdf")
-loader = PyPDFLoader(pdf_path)
+# Load PDF
+loader = PyPDFLoader("chimera.pdf")
 documents = loader.load()
 
-# --- 2️⃣ Split into chunks ---
+# Split documents
 text_splitter = CharacterTextSplitter(chunk_size=800, chunk_overlap=100)
 texts = text_splitter.split_documents(documents)
 
-# --- 3️⃣ Create embeddings + FAISS vector store ---
+# Embeddings
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 db = FAISS.from_documents(texts, embeddings)
 
-# --- 4️⃣ Build retriever-based chatbot ---
 retriever = db.as_retriever(search_kwargs={"k": 3})
 
-# --- 5️⃣ Hugging Face LLM setup using secret ---
-hf_token = os.environ["HUGGINGFACEHUB_API_TOKEN"]  # Must be set in Space Secrets
+# Hugging Face Hub LLM
+hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+if hf_token is None:
+    raise ValueError("HUGGINGFACEHUB_API_TOKEN not set in Space Secrets!")
 
 llm = HuggingFaceHub(
     repo_id="google/flan-t5-base",
@@ -40,82 +37,48 @@ qa = ConversationalRetrievalChain.from_llm(
     retriever=retriever
 )
 
-# --- 6️⃣ Chat history ---
 chat_history = []
 
-# --- 7️⃣ Respond function ---
 def respond(message, history):
-    history = history[-6:]  # Keep last 3 exchanges
+    history = history[-6:]
     result = qa({"question": message, "chat_history": history})
     history.append((message, result["answer"]))
     return history, history
 
-# --- 8️⃣ Gradio UI with Entry Warning ---
 with gr.Blocks() as demo:
     with gr.Column():
-        # Warning message
         warning_text = gr.HTML(
-            """
-            <div style="background-color:black;color:white;padding:20px;font-family:monospace;font-size:18px;">
-            ⚠ WARNING — INVESTIGATIVE SIMULATION ⚠<br><br>
-            You are about to enter <b>The Chimera Case</b>, a high-stakes investigation into Innovate Future Labs (IFL) and Project Chimera.<br>
-            The scenario contains allegations, leaked files, and disputed testimonies. Treat every claim as unverified until verified by evidence.<br>
-            Your decisions and observations will guide your understanding of the case.<br><br>
-            Are you ready to proceed?
-            </div>
-            """
+            "<div style='background-color:black;color:white;padding:20px;'>⚠ WARNING: Investigative Simulation ⚠<br>Are you ready?</div>"
         )
-
-        # Buttons
         enter_btn = gr.Button("Enter the Case")
         exit_btn = gr.Button("Exit")
-
-        # Chatbot (hidden initially)
         chatbot = gr.Chatbot(visible=False)
-        user_input = gr.Textbox(placeholder="Type your message here...", visible=False)
+        user_input = gr.Textbox(placeholder="Type here...", visible=False)
         submit_btn = gr.Button("Send", visible=False)
 
-    # --- Button interactions ---
     def enter_case():
         return (
-            gr.update(visible=True),  # chatbot
-            gr.update(visible=True),  # user_input
-            gr.update(visible=True),  # submit_btn
-            gr.update(value=""),      # hide warning
-            gr.update(visible=False), # hide enter_btn
-            gr.update(visible=False)  # hide exit_btn
+            gr.update(visible=True),
+            gr.update(visible=True),
+            gr.update(visible=True),
+            gr.update(value=""),
+            gr.update(visible=False),
+            gr.update(visible=False)
         )
 
     def exit_case():
         return (
-            gr.update(value="<h2>Session ended. You exited the simulation.</h2>"),  # hide warning
-            gr.update(visible=False),  # chatbot
-            gr.update(visible=False),  # user_input
-            gr.update(visible=False),  # submit_btn
-            gr.update(visible=False),  # enter_btn
-            gr.update(visible=False)   # exit_btn
+            gr.update(value="Session ended."),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False)
         )
 
-    # Connect buttons
-    enter_btn.click(
-        enter_case,
-        inputs=None,
-        outputs=[chatbot, user_input, submit_btn, warning_text, enter_btn, exit_btn]
-    )
+    enter_btn.click(enter_case, inputs=None, outputs=[chatbot, user_input, submit_btn, warning_text, enter_btn, exit_btn])
+    exit_btn.click(exit_case, inputs=None, outputs=[warning_text, chatbot, user_input, submit_btn, enter_btn, exit_btn])
+    submit_btn.click(respond, inputs=[user_input, chatbot], outputs=[chatbot, chatbot])
 
-    exit_btn.click(
-        exit_case,
-        inputs=None,
-        outputs=[warning_text, chatbot, user_input, submit_btn, enter_btn, exit_btn]
-    )
-
-    submit_btn.click(
-        respond,
-        inputs=[user_input, chatbot],
-        outputs=[chatbot, chatbot]
-    )
-
-# --- 9️⃣ Launch ---
 if __name__ == "__main__":
     demo.launch(share=True, enable_queue=True)
-