Create app.py

app.py

@@ -0,0 +1,119 @@
+import gradio as gr
+from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
+import os
+
+# Load and index Paul Graham documents
+pg_docs = SimpleDirectoryReader("data/paul").load_data()
+pg_docs = [doc for doc in pg_docs if doc.text and doc.text.strip()]
+pg_index = VectorStoreIndex.from_documents(pg_docs)
+pg_engine = pg_index.as_query_engine()
+
+# Load and index Insurance documents
+ins_docs = SimpleDirectoryReader("data/insurance").load_data()
+ins_docs = [doc for doc in ins_docs if doc.text and doc.text.strip()]
+ins_index = VectorStoreIndex.from_documents(ins_docs)
+ins_engine = ins_index.as_query_engine()
+
+# Query functions with input validation
+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?"
+]
+
+# UI
+def launch_interface():
+    with gr.Blocks(
+        title="RAG App",
+        css="""
+        .gradio-container {
+            background-color: #e6fff7 !important;
+        }
+        #header-text {
+            text-align: center;
+            color: #2b6777;
+        }
+        """
+    ) as demo:
+
+        gr.Markdown("""
+            <div id='header-text'>
+                <h1>RAG Bot with LlamaIndex</h1>
+            </div>
+        """)
+
+        with gr.Tabs():
+            # Paul Graham Tab
+            with gr.Tab("Paul Graham"):
+                if os.path.exists("data/logo.png"):
+                    gr.Image("data/logo.png", show_label=False, container=False, height=120)
+
+                gr.Markdown("""
+                    <div id='header-text'>
+                        <h2>Paul Graham: Writing and Thinking</h2>
+                        <p>Select a question or ask your own based on the essay.</p>
+                    </div>
+                """)
+
+                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)
+                clear_pg.click(lambda: ("", "", ""), outputs=[dropdown_pg, textbox_pg, output_pg])
+
+            # Insurance Tab
+            with gr.Tab("Insurance"):
+                gr.Markdown("""
+                    <div id='header-text'>
+                        <h2>Understanding Insurance</h2>
+                        <p>Explore key insurance concepts or ask your own questions.</p>
+                    </div>
+                """)
+
+                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)
+                clear_ins.click(lambda: ("", "", ""), outputs=[dropdown_ins, textbox_ins, output_ins])
+
+    demo.launch()
+
+if __name__ == "__main__":
+    launch_interface()