Update app.py

app.py

@@ -1,70 +1,3 @@
-#pip install llama-index-embeddings-huggingface
-import gradio as gr
-import os
-from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
-from llama_index.embeddings.huggingface import HuggingFaceEmbedding
-
-
-# ✅ Access OpenAI API Key
-openai_api_key = os.environ.get("OPENAI_API_KEY")
-if not openai_api_key:
-    raise ValueError("❌ OPENAI_API_KEY not found. Add it in Space settings > Secrets.")
-os.environ["OPENAI_API_KEY"] = openai_api_key
-
-# ✅ Set Hugging Face Embedding globally via Settings
-embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
-Settings.embed_model = embed_model  # Replaces deprecated ServiceContext
-
-# ✅ Helper to load and filter documents
-def load_filtered_docs(folder):
-    docs = SimpleDirectoryReader(folder).load_data()
-    return [doc for doc in docs if doc.text and doc.text.strip()]
-
-# ✅ Load Paul Graham documents
-pg_docs = load_filtered_docs("data/paul")
-pg_index = VectorStoreIndex.from_documents(pg_docs)
-pg_engine = pg_index.as_query_engine()
-
-# ✅ Load Insurance documents
-ins_docs = load_filtered_docs("data/insurance")
-ins_index = VectorStoreIndex.from_documents(ins_docs)
-ins_engine = ins_index.as_query_engine()
-
-# ✅ Query functions
-def query_pg(query):
-    if not query.strip():
-        return "❌ Please enter a valid question before submitting."
-    try:
-        return str(pg_engine.query(query))
-    except Exception as e:
-        return f"❌ Error: {str(e)}"
-
-def query_ins(query):
-    if not query.strip():
-        return "❌ Please enter a valid question before submitting."
-    try:
-        return str(ins_engine.query(query))
-    except Exception as e:
-        return f"❌ Error: {str(e)}"
-
-# ✅ Predefined questions
-paul_questions = [
-    "What is the main purpose of writing, according to Paul Graham?",
-    "Why do students often struggle with writing in school?",
-    "How does Paul Graham describe the relationship between writing and thinking?",
-    "What is one reason Paul Graham gives for why school essays feel boring?",
-    "What does Paul Graham suggest writers should focus on first?"
-]
-
-insurance_questions = [
-    "What is insurance and why is it important?",
-    "What types of insurance are common?",
-    "How does life insurance work?",
-    "What is the difference between premium and coverage?",
-    "What should you check before buying insurance?"
-]
-
-# ✅ Gradio Interface
 def launch_interface():
     with gr.Blocks(
         title="RAG App",
@@ -79,14 +12,11 @@ def launch_interface():
         """
     ) as demo:
 
-        gr.Markdown("""
-            <div id='header-text'>
-                <h1>RAG Bot with LlamaIndex (PDF + TXT)</h1>
-            </div>
-        """)
+        gr.Markdown("""<div id='header-text'><h1>RAG Bot with LlamaIndex (PDF + TXT)</h1></div>""")
 
         with gr.Tabs():
             with gr.Tab("Paul Graham"):
+                # ✅ Logo moved above heading
                 if os.path.exists("data/logo.png"):
                     gr.Image("data/logo.png", show_label=False, container=False, height=120)
 
@@ -97,15 +27,20 @@ def launch_interface():
                     </div>
                 """)
 
-                dropdown_pg = gr.Dropdown(label="Pick a Question", choices=paul_questions, interactive=True)
+                dropdown_pg = gr.Dropdown(label="Pick a Question", choices=[""] + paul_questions, interactive=True)
                 textbox_pg = gr.Textbox(label="Ask Anything", placeholder="Type your question...", lines=2)
                 output_pg = gr.Textbox(label="Response", lines=10)
-
                 submit_pg = gr.Button("Submit")
                 clear_pg = gr.Button("Clear")
 
-                dropdown_pg.change(query_pg, inputs=dropdown_pg, outputs=output_pg)
-                submit_pg.click(query_pg, inputs=textbox_pg, outputs=output_pg)
+                def handle_pg_submit(drop_value, text_value):
+                    final_query = drop_value if drop_value else text_value
+                    if not final_query.strip():
+                        return "❌ Please select or enter a question."
+                    return query_pg(final_query)
+
+                dropdown_pg.change(fn=query_pg, inputs=dropdown_pg, outputs=output_pg)
+                submit_pg.click(handle_pg_submit, inputs=[dropdown_pg, textbox_pg], outputs=output_pg)
                 clear_pg.click(lambda: ("", "", ""), outputs=[dropdown_pg, textbox_pg, output_pg])
 
             with gr.Tab("Insurance"):
@@ -116,18 +51,20 @@ def launch_interface():
                     </div>
                 """)
 
-                dropdown_ins = gr.Dropdown(label="Pick a Question", choices=insurance_questions, interactive=True)
+                dropdown_ins = gr.Dropdown(label="Pick a Question", choices=[""] + insurance_questions, interactive=True)
                 textbox_ins = gr.Textbox(label="Ask Anything", placeholder="Type your question...", lines=2)
                 output_ins = gr.Textbox(label="Response", lines=10)
-
                 submit_ins = gr.Button("Submit")
                 clear_ins = gr.Button("Clear")
 
-                dropdown_ins.change(query_ins, inputs=dropdown_ins, outputs=output_ins)
-                submit_ins.click(query_ins, inputs=textbox_ins, outputs=output_ins)
+                def handle_ins_submit(drop_value, text_value):
+                    final_query = drop_value if drop_value else text_value
+                    if not final_query.strip():
+                        return "❌ Please select or enter a question."
+                    return query_ins(final_query)
+
+                dropdown_ins.change(fn=query_ins, inputs=dropdown_ins, outputs=output_ins)
+                submit_ins.click(handle_ins_submit, inputs=[dropdown_ins, textbox_ins], outputs=output_ins)
                 clear_ins.click(lambda: ("", "", ""), outputs=[dropdown_ins, textbox_ins, output_ins])
 
     demo.launch()
-
-if __name__ == "__main__":
-    launch_interface()