Updated background agents latency checks

agent.py

@@ -8,7 +8,6 @@ from github import Github, Auth
 from langchain_groq import ChatGroq
 from langchain_core.tools import tool
 from langchain.agents import create_tool_calling_agent, AgentExecutor
-from langchain import hub
 
 # Load environment variables
 load_dotenv()

app.py

@@ -3,8 +3,8 @@ import time
 import threading
 import logging
 from gradio.themes.utils import sizes
-from main import run_repository_ranking  # Import the repository ranking function
-import agent  # Import the test.py module for chat agent
+from main import run_repository_ranking
+import agent 
 
 # ---------------------------
 # Global Logging Buffer Setup
@@ -218,14 +218,10 @@ with gr.Blocks(
     def chat_with_agent(user_msg, repo, history):
         logging.info("[Chat] User sent message: '%s' for repo: '%s'", user_msg, repo)
         if not user_msg or not user_msg.strip():
-            yield history + [["", "Please enter a message before sending."]], history
-            return
+            # Block blank messages
+            return history + [["", "Please enter a message before sending."]], history
         if not repo:
-            yield history + [[user_msg, "Please select a repository first."]], history
-            return
-        # Show user message immediately with a loading placeholder
-        new_history = history + [[user_msg, "⏳ Loading..."]]
-        yield new_history, new_history
+            return history + [[user_msg, "Please select a repository first."]], history
         full_query = f"[{repo}] {user_msg}"
         try:
             result = agent.agent_executor.invoke({"input": full_query})
@@ -234,9 +230,8 @@ with gr.Blocks(
         except Exception as e:
             answer = f"Error: {e}"
             logging.error("[Chat] Error in agent_executor: %s", e)
-        # Replace the loading message with the actual answer
-        new_history[-1][1] = answer
-        yield new_history, new_history
+        history = history + [[user_msg, answer]]
+        return history, history
 
     # Disable send button if no repo is selected or message is blank, and show a helpful message
     def can_send(user_msg, repo):