Update app.py

app.py

@@ -1,52 +1,34 @@
-import gradio as gr
-from policy_retriever import PolicyRetriever
-
-# Initialize the retriever
-retriever = PolicyRetriever()
-
-def search(query):
-    results = retriever.search(query)
-    for result in results:
-        if len(result['content'].strip()) > 50:
-            return f"""
-            Search Results for: '{query}'
-            
-            Relevance Score: {result['score']:.3f}
-            
-            Section:
-            {'-' * 40}
-            {result['heading']}
-            
-            Content:
-            {'-' * 40}
-            {result['content']}
-            
-            {'=' * 80}
-            """
-    return "No relevant results found."
-
-# Create the Gradio interface
-demo = gr.Interface(
-    fn=search,
-    inputs=[
-        gr.Textbox(
-            label="Enter your question about the insurance policy",
-            placeholder="e.g., What are the liability limits?",
-            lines=2
-        )
-    ],
-    outputs=gr.Textbox(label="Policy Information", lines=10),
-    title="Insurance Policy Search",
-    description="""
-    Ask any question about your insurance policy!
-    """,
-    examples=[
-        ["What are the liability limits?"],
-        ["How does insurance work with multiple vehicles?"],
-        ["What happens if there is other insurance?"],
-        ["How do medical payments work?"]
-    ]
-)
-
-if __name__ == "__main__":
-    demo.launch()
+import gradio as gr
+from policy_retriever_2 import PolicyRetriever
+
+# Initialize the PolicyRetriever
+retriever = PolicyRetriever()
+
+def process_query(query):
+    """Process a query and return the answer"""
+    if not query.strip():
+        return "Please enter a question about the policy."
+    
+    try:
+        answer = retriever.search_and_generate(query)
+        return answer
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
+# Create the Gradio interface
+demo = gr.Interface(
+    fn=process_query,
+    inputs=gr.Textbox(label="Ask a question about your policy", placeholder="e.g., What are the liability limits?"),
+    outputs=gr.Textbox(label="Answer"),
+    title="Insurance Policy Q&A",
+    description="Ask questions about your insurance policy and get answers based on the policy document.",
+    examples=[
+        ["What are the liability limits for bodily injury and property damage?"],
+        ["Who is considered an insured person under this policy?"],
+        ["What types of vehicles are covered under insured autos?"],
+        ["What is excluded from liability coverage?"]
+    ]
+)
+
+if __name__ == "__main__":
+    demo.launch()