Update app.py

app.py

@@ -1,26 +1,36 @@
 import gradio as gr
-from inference import get_evo_response, get_gpt_response
+from inference import get_evo_response, get_gpt_response, get_context_from_file
+from logger import log_feedback
 
-def advisor_interface(prompt, option1, option2):
-    evo_result = get_evo_response(prompt, option1, option2)
-    gpt_result = get_gpt_response(prompt, option1, option2)
-    return evo_result, gpt_result
+def advisor_interface(query, file, feedback_choice):
+    context = None
+    if file is not None:
+        context = get_context_from_file(file)
+
+    evo_output = get_evo_response(query, file)
+    gpt_output = get_gpt_response(query, file)
+
+    if feedback_choice != "No feedback":
+        log_feedback(query, context, evo_output, feedback_choice)
+
+    return evo_output, gpt_output
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 EvoRAG – Retrieval-Augmented Adaptive AI for Reasoning")
-    gr.Markdown("Ask a commonsense or financial reasoning question with two options. Evo and GPT-3.5 will both suggest the better one.")
+    gr.Markdown("## 🧠 EvoRAG – Retrieval-Augmented Adaptive AI for Finance")
 
     with gr.Row():
-        prompt = gr.Textbox(label="📝 Your Question", placeholder="e.g. Should we reduce exposure to Tech Fund A this quarter?")
+        query = gr.Textbox(label="📝 Ask a financial question", placeholder="e.g. Should we reduce exposure to Fund A?")
+    
+    file = gr.File(label="📂 Upload policy or memo (.pdf or .txt)", type="file")
+
     with gr.Row():
-        option1 = gr.Textbox(label="🔹 Option 1", placeholder="Yes, reduce exposure this quarter.")
-        option2 = gr.Textbox(label="🔸 Option 2", placeholder="No, maintain current allocation.")
+        feedback = gr.Radio(["👍 Helpful", "👎 Not Helpful", "No feedback"], label="Was Evo’s answer useful?", value="No feedback")
 
     with gr.Row():
-        evo_output = gr.Textbox(label="🔬 EvoRAG Suggestion")
-        gpt_output = gr.Textbox(label="🤖 GPT-3.5 Suggestion")
+        evo_out = gr.Textbox(label="🔬 EvoRAG Suggestion")
+        gpt_out = gr.Textbox(label="🤖 GPT-3.5 Suggestion")
 
-    run_btn = gr.Button("Run Advisors")
-    run_btn.click(fn=advisor_interface, inputs=[prompt, option1, option2], outputs=[evo_output, gpt_output])
+    submit_btn = gr.Button("Run Advisors")
+    submit_btn.click(fn=advisor_interface, inputs=[query, file, feedback], outputs=[evo_out, gpt_out])
 
 demo.launch()