Update tools.py

tools.py

@@ -42,6 +42,8 @@ from PIL import Image
 from smolagents import DuckDuckGoSearchTool
 
 
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+
 
 def web_search(query, max_results=3):
     searchtool = DuckDuckGoSearchTool(query)
@@ -49,8 +51,18 @@ def web_search(query, max_results=3):
     return result
     
 def wiki_search(query):
-    wikipedia.set_lang("en")
-    return wikipedia.summary(query, sentences=5)
+    #wikipedia.set_lang("en")
+    #return wikipedia.summary(query, sentences=5)
+
+    model = "mistralai/Mistral-7B-Instruct-v0.2"  # or another free model
+    tokenizer = AutoTokenizer.from_pretrained(model)
+    model = AutoModelForCausalLM.from_pretrained(model)
+    
+    chat = pipeline("text-generation", model=model, tokenizer=tokenizer)
+    
+    response = chat(query, max_new_tokens=50)
+    return response[0]['generated_text']
+
 
 def python_repl(code):
     code = "AkylaiBva/Final_Assignment_Template_AB/"+code
@@ -78,8 +90,7 @@ def speech_recognition(path):
     return r.recognize_google(audio)
 
 def reverse_string(s):
-    rs = s[::-1]
-    return web_search(rs)
+    return s[::-1]
 
 def query_image(path):
     path = "AkylaiBva/Final_Assignment_Template_AB/"+ path