Update app.py

app.py

@@ -51,81 +51,61 @@ def query_tqa(query, search_level):
     # )
     print(str(grag_response.response))
     return (
-        str(grag_response.response),
-        # grag_reference,
-        # grag_reference_text,
-        # rag_response,
-        # rag_reference,
-        # rag_reference_text,
+        str(grag_response.response)
     )
 
 
-def show_graph():
-    """
-    Show the latest graph visualization in an iframe.
-
-    Returns:
-    str: The HTML content to display the graph visualization in an iframe.
-    """
-
-    graph_vis_dir = os.getenv("GRAPH_VIS", "graph_vis")
-    try:
-        latest_graph = get_latest_html_file(graph_vis_dir)
-        if latest_graph:
-            with open(latest_graph, "r", encoding="utf-8") as f:
-                html_content = f.read()
 
-            encoded_html = base64.b64encode(html_content.encode()).decode()
-            iframe_html = f'<iframe src="data:text/html;base64,{encoded_html}" width="100%" height="1000px" frameborder="0"></iframe>'
-            return iframe_html
-        else:
-            return "No graph visualization found."
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-
-def reveal_coupon(query, grag_response):
-    """
-    Get the coupon from the query and response.
+ 
+# with gr.Blocks() as demo:
+#     gr.Markdown("# Comfy Virtual Assistant")
+#     chatbot = gr.Chatbot(
+#         label="Comfy Virtual Assistant",
+#         type="messages",
+#         scale=1,
+#         # suggestions = [
+#         #         {"text": "How much iphone cost?"},
+#         #         {"text": "What phone options do i have ?"}
+#         #         ],
+           
+#     )
+#     msg = gr.Textbox(label="Input Your Query")
+#     clear = gr.ClearButton([msg, chatbot])
+
+#     def respond(message, chat_history):
+#             bot_message = query_tqa(message, 2)
+#             # chat_history.append((message, bot_message))
+#             chat_history.append(ChatMessage(role="user", content=message))
+#             chat_history.append(ChatMessage(role="assistant", content=bot_message))
+#             time.sleep(1)
+#             return "", chat_history
+    
+#     msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-    Args:
-    query (str): Query asked to Graph-RAG.
-    grag_response (str): Response from the Graph-RAG model.
+def chatbot_response(message: str, history: List[Tuple[str, str]]) -> str:
+    # Use the query_tqa function to get the response
+    search_level = 2  # You can adjust this or make it configurable
+    response = query_tqa(message, search_level)
+    return response
 
-    Returns:
-    str: Coupon with reasoning.
-    """
+# Create the Gradio interface
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
 
-    if not query.strip() or not grag_response.strip():
-        raise gr.Error("Please ask a query and get a response before revealing the coupon.")
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
 
-    coupon = get_coupon(query, grag_response)
-    return coupon
+    def bot(history):
+        bot_message = chatbot_response(history[-1][0], history)
+        history[-1][1] = bot_message
+        return history
 
- 
-with gr.Blocks() as demo:
-    gr.Markdown("# Comfy Virtual Assistant")
-    chatbot = gr.Chatbot(
-        label="Comfy Virtual Assistant",
-        type="messages",
-        scale=1,
-        # suggestions = [
-        #         {"text": "How much iphone cost?"},
-        #         {"text": "What phone options do i have ?"}
-        #         ],
-           
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
     )
-    msg = gr.Textbox(label="Input Your Query")
-    clear = gr.ClearButton([msg, chatbot])
-
-    def respond(message, chat_history):
-            bot_message = query_tqa(message, 2)
-            # chat_history.append((message, bot_message))
-            chat_history.append(ChatMessage(role="user", content=message))
-            chat_history.append(ChatMessage(role="assistant", content=bot_message))
-            time.sleep(1)
-            return "", chat_history
+    clear.click(lambda: None, None, chatbot, queue=False)
+
     
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-     
 demo.launch(auth=(os.getenv("ID"), os.getenv("PASS")), share=False)