Update app.py

app.py

@@ -1,3 +1,67 @@
+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?"
+]
 def launch_interface():
     with gr.Blocks(
         title="RAG App",
@@ -68,3 +132,7 @@ def launch_interface():
                 clear_ins.click(lambda: ("", "", ""), outputs=[dropdown_ins, textbox_ins, output_ins])
 
     demo.launch()
+
+
+if __name__ == "__main__":
+    launch_interface()