Update app.py

app.py

@@ -135,9 +135,16 @@ import numpy as np
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 import gradio as gr
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import nltk
 
+import datetime
+import requests
+import pytz
+from tools.final_answer import FinalAnswerTool
+
+from Gradio_UI import GradioUI
+
 nltk.download("stopwords")
 from nltk.corpus import stopwords
 
@@ -226,6 +233,10 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+
 # Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -303,17 +314,20 @@ def search_papers(user_input):
     print("DEBUG: No results found.")
     return "No results found. Try different keywords."
 
+# Launch Gradio UI with CodeAgent
+GradioUI(agent).launch()
+
 
-# Create Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("# ScholarAgent")
-    keyword_input = gr.Textbox(label="Enter keywords (comma-separated)", placeholder="e.g., deep learning, reinforcement learning")
-    output_display = gr.Markdown()
-    search_button = gr.Button("Search")
+# # Create Gradio UI
+# with gr.Blocks() as demo:
+#     gr.Markdown("# ScholarAgent")
+#     keyword_input = gr.Textbox(label="Enter keywords (comma-separated)", placeholder="e.g., deep learning, reinforcement learning")
+#     output_display = gr.Markdown()
+#     search_button = gr.Button("Search")
 
-    search_button.click(search_papers, inputs=[keyword_input], outputs=[output_display])
+#     search_button.click(search_papers, inputs=[keyword_input], outputs=[output_display])
 
-    print("DEBUG: Gradio UI is running. Waiting for user input...")
+#     print("DEBUG: Gradio UI is running. Waiting for user input...")
 
-# Launch Gradio App
-demo.launch()
+# # Launch Gradio App
+# demo.launch()