Update app.py

solved problem in call_llm

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 import json
 import hashlib
 import numpy as np
-from typing import Optional, List, Tuple, ClassVar, Dict
+from typing import Optional, List, Dict
 
 from sentence_transformers import SentenceTransformer, util, CrossEncoder
 from langchain.llms.base import LLM
@@ -50,9 +50,6 @@ def call_llm(prompt: str) -> str:
     """
     try:
         result = llm(prompt)  # Directly call the model with the prompt
-        # If the result is a dictionary and contains 'text', extract it.
-        if isinstance(result, dict) and 'text' in result:
-            return result['text']
         return result if isinstance(result, str) else str(result)
     except Exception as e:
         logger.error(f"LLM call error: {e}")
@@ -179,10 +176,8 @@ sanity_checker = QuestionSanityChecker(llm)
 ###############################################################################
 # 7) smolagents Integration: GROQ Model and Web Search
 ###############################################################################
-# Instantiate the DuckDuckGo search tool
 search_tool = DuckDuckGoSearchTool()
 
-# Create the web agent with the search tool
 web_agent = CodeAgent(
     tools=[search_tool],
     model=llm