Update app.py

app.py

@@ -14,13 +14,11 @@ import numpy as np
 import ast
 import operator as op
 
-# === Setup ===
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
-
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# === LLM Setup (Shared) ===
+
 llm = HuggingFaceLLM(
     context_window=4096,
     max_new_tokens=512,
@@ -29,7 +27,7 @@ llm = HuggingFaceLLM(
     model_name="Gensyn/Qwen2.5-0.5B-Instruct",
 )
 
-# === Question Validation Component ===
+
 class QuestionValidation:
     def __init__(self, llm_client):
         self.client = llm_client
@@ -51,7 +49,7 @@ class QuestionValidation:
             "similarity": round(float(similarity), 4)
         }
 
-# === Tools ===
+
 def search_web(query: str, max_results: int = 5) -> List[Dict[str, str]]:
     try:
         with DDGS() as ddgs:
@@ -89,7 +87,7 @@ def evaluate_math_expression(expr: str) -> str:
     except Exception as e:
         return f"Error evaluating expression: {e}"
 
-# Instantiate validator using shared LLM
+
 validator = QuestionValidation(llm)
 
 validate_tool = FunctionTool.from_defaults(
@@ -119,47 +117,17 @@ agent = ReActAgent.from_tools(
     max_iterations=3
 )
 
-# === Question Loop Logic ===
-def question_loop_agent(user_question: str):
-    llm_answer = llm.complete(user_question).text.strip()
-    similarity_score = 0.0
-    retry = 0
-    max_retries = 2
-
-    while retry < max_retries:
-        guessed_question = validator.guess_question(llm_answer)
-        similarity_score = validator.compute_similarity(user_question, guessed_question)
-        if similarity_score > 0.6:
-            break
-        retry += 1
-
-    return (
-        f"Original Question: {user_question}\n\n"
-        f"Answer: {llm_answer}\n\n"
-        f"Guessed Question: {guessed_question}\n\n"
-        f"Similarity Score: {similarity_score:.4f}"
-    )
+def respond(message: str, history: List[List[str]]) -> str:
+    response = agent.chat(message)
+    return response.response
 
-# === Gradio Interfaces ===
 with gr.Blocks() as app:
-    with gr.Tab("Validation Loop"):
-        gr.Interface(
-            fn=question_loop_agent,
-            inputs=gr.Textbox(lines=2, placeholder="Ask me a question..."),
-            outputs="text",
-            title="Question Similarity Loop Agent",
-            description="Loops until the guessed question has a similarity score > 0.6."
-        ).render()
-
-        gr.ChatInterface(
-            respond,
-            additional_inputs=[
-                gr.Textbox(value="You are a Chatbot.", label="System message"),
-                gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-                gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-                gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
-            ],
-        ).render()
+    gr.ChatInterface(
+        respond,
+        chatbot=gr.Chatbot(),
+        title="Smart Assistant",
+        description="Ask me anything — math, web search, or validating a question.",
+    )
 
 if __name__ == "__main__":
     app.launch()