Update app.py

app.py

@@ -259,7 +259,7 @@ def generate_answer(message, choice, retrieval_mode):
 
     prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
 
-    if retrieval_mode == "Vector":
+    if retrieval_mode == "VDB":
         qa_chain = RetrievalQA.from_chain_type(
             llm=chat_model,
             chain_type="stuff",
@@ -269,7 +269,7 @@ def generate_answer(message, choice, retrieval_mode):
         response = qa_chain({"query": message})
         logging.debug(f"Vector response: {response}")
         return response['result'], extract_addresses(response['result'])
-    elif retrieval_mode == "Knowledge-Graph":
+    elif retrieval_mode == "KGF":
         response = chain_neo4j.invoke({"question": message})
         logging.debug(f"Knowledge-Graph response: {response}")
         return response, extract_addresses(response)
@@ -795,7 +795,7 @@ def fetch_local_weather():
 
 
 def handle_retrieval_mode_change(choice):
-    if choice == "Knowledge-Graph":
+    if choice == "KGF":
         return gr.update(interactive=False), gr.update(interactive=False)
     else:
         return gr.update(interactive=True), gr.update(interactive=True)
@@ -808,7 +808,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
 
             chatbot = gr.Chatbot([], elem_id="RADAR:Channel 94.1", bubble_full_width=False)
             choice = gr.Radio(label="Select Style", choices=["Details", "Conversational"], value="Conversational")
-            retrieval_mode = gr.Radio(label="Retrieval Mode", choices=["Vector", "Knowledge-Graph"], value="Vector")
+            retrieval_mode = gr.Radio(label="Retrieval Mode", choices=["VDB", "KGF"], value="VDB")
 
             gr.Markdown("<h1 style='color: red;'>Talk to RADAR</h1>", elem_id="voice-markdown")
 
@@ -854,10 +854,10 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             # Handle retrieval mode change
             retrieval_mode.change(fn=handle_retrieval_mode_change, inputs=retrieval_mode, outputs=[choice, choice])
 
-        # with gr.Column():
-        #     weather_output = gr.HTML(value=fetch_local_weather())
-        #     news_output = gr.HTML(value=fetch_local_news())
-        #     events_output = gr.HTML(value=fetch_local_events())            
+        with gr.Column():
+            weather_output = gr.HTML(value=fetch_local_weather())
+            news_output = gr.HTML(value=fetch_local_news())
+            events_output = gr.HTML(value=fetch_local_events())            
         
 
         with gr.Column():