Update app.py

app.py

@@ -3,7 +3,6 @@ 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:
@@ -12,19 +11,22 @@ 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
+Settings.embed_model = embed_model
 
-# ✅ Helper to load and filter documents
+# ✅ 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()]
+    try:
+        docs = SimpleDirectoryReader(folder).load_data()
+        return [doc for doc in docs if doc.text and doc.text.strip()]
+    except Exception as e:
+        print(f"❌ Error loading docs from {folder}: {e}")
+        return []
 
-# ✅ Load Paul Graham documents
+# ✅ Load Paul Graham and Insurance 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()
@@ -62,6 +64,8 @@ insurance_questions = [
     "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",
@@ -80,7 +84,6 @@ def launch_interface():
 
         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)
 
@@ -103,13 +106,12 @@ def launch_interface():
                         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)
-               def handle_pg_clear():
-                   return "", "", ""
-
-               clear_pg.click(fn=handle_pg_clear, outputs=[dropdown_pg, textbox_pg, output_pg])
+                def handle_pg_clear():
+                    return "", "", ""
 
+                dropdown_pg.change(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(fn=handle_pg_clear, outputs=[dropdown_pg, textbox_pg, output_pg])
 
             with gr.Tab("Insurance"):
                 gr.Markdown("""
@@ -131,16 +133,14 @@ def launch_interface():
                         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)
                 def handle_ins_clear():
-                     return "", "", ""
+                    return "", "", ""
 
+                dropdown_ins.change(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(fn=handle_ins_clear, outputs=[dropdown_ins, textbox_ins, output_ins])
 
-
     demo.launch()
 
-
-if __name__ == "__main__":
-    launch_interface()
+# ✅ Launch app (for Spaces)
+launch_interface()