Update app.py

app.py

@@ -1,84 +1,44 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import GradioUI
 import datetime
-import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
-from tools.visit_webpage import VisitWebpageTool
-#from tools.web_shearch import DuckDuckGoSearchTool
-from smolagents import GradioUI
 import tempfile
 import gradio as gr
 import os
-from tools.create_presentation import create_presentation
+import re
 
+from tools.final_answer import FinalAnswerTool
+from tools.visit_webpage import VisitWebpageTool
+from tools.create_presentation import create_presentation
 
-'''
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def my_custom_tool(x:str, y:int)-> int: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-'''
+# =========================
+# DOCUMENT CREATION TOOL
+# =========================
 @tool
 def create_document(text: str, format: str = "docx") -> str:
-    """Creates a document with the provided text and allows download.
-    
-    Args:
-        text: The text content to write to the document
-        format: The output format, either 'docx' or 'pdf'
-    """
+    """Creates a document with the provided text and allows download."""
     try:
         import docx
         from docx.shared import Pt
-        
-        # Create a temp directory to store files
+
         temp_dir = tempfile.mkdtemp()
-        
-        # Create a new document
         doc = docx.Document()
-        
-        # Add a heading
-        doc.add_heading('Generated Document', 0)
-        
-        # Set font style for regular text
-        style = doc.styles['Normal']
+        doc.add_heading("Generated Document", 0)
+
+        style = doc.styles["Normal"]
         font = style.font
-        font.name = 'Calibri'
+        font.name = "Calibri"
         font.size = Pt(11)
-        
-        # Add paragraphs from the input text
-        # Split by newlines to maintain paragraph structure
-        for paragraph in text.split('\n'):
-            if paragraph.strip():  # Skip empty paragraphs
+
+        for paragraph in text.split("\n"):
+            if paragraph.strip():
                 doc.add_paragraph(paragraph)
-        
-        # Save the document
+
         docx_path = os.path.join(temp_dir, "generated_document.docx")
         doc.save(docx_path)
-        
-        # Convert to PDF if requested
+
         if format.lower() == "pdf":
             try:
                 from docx2pdf import convert
@@ -86,124 +46,103 @@ def create_document(text: str, format: str = "docx") -> str:
                 convert(docx_path, pdf_path)
                 return pdf_path
             except ImportError:
-                return f"PDF conversion requires docx2pdf package. Document saved as DOCX instead at: {docx_path}"
-        
+                return docx_path
+
         return docx_path
-    
+
     except Exception as e:
         return f"Error creating document: {str(e)}"
 
-# Custom file download tool to help with file handling
-@tool
-def get_file_download_link(file_path: str) -> str:
-    """Creates a download link for a file.
-    
-    Args:
-        file_path: Path to the file that should be made available for download
-    """
-    if not os.path.exists(file_path):
-        return f"Error: File not found at {file_path}"
-    
-    # Get file extension and set up appropriate mime type
-    _, file_extension = os.path.splitext(file_path)
-    mime_types = {
-        '.docx': 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
-        '.pdf': 'application/pdf',
-    }
-    mime_type = mime_types.get(file_extension.lower(), 'application/octet-stream')
-    
-    # Return information that can be used by the agent to instruct the user
-    return f"File ready for download: {os.path.basename(file_path)} ({mime_type})"
-        
 
+# =========================
+# AGENT SETUP
+# =========================
 final_answer = FinalAnswerTool()
-#web_search=DuckDuckGoSearchTool()
-visit_webpage=VisitWebpageTool()
-
-# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
-# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
+visit_webpage = VisitWebpageTool()
 
-# Load LLM
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen3-235B-A22B', # it is possible that this model may be overloaded
-    custom_role_conversions=None,
+    model_id="Qwen/Qwen3-235B-A22B",
 )
 
-
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-with open("prompts.yaml", 'r') as stream:
+with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-#,web_search
-    
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,visit_webpage,create_document,get_file_download_link,create_presentation,], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        visit_webpage,
+        create_document,
+        create_presentation,   # ✅ PPTX tool
+    ],
     max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
+    verbosity_level=0,
+    prompt_templates=prompt_templates,
 )
 
-# Custom Gradio UI with file download capability
+
+# =========================
+# CUSTOM GRADIO UI
+# =========================
 class CustomGradioUI(GradioUI):
     def build_interface(self):
         with gr.Blocks() as interface:
-            with gr.Row():
-                with gr.Column(scale=1):
-                    gr.Markdown("# AI Assistant")
-                
+            gr.Markdown("# AI Assistant")
+
             chatbot = gr.Chatbot(height=600)
             msg = gr.Textbox(
-                placeholder="Ask me anything...",
+                placeholder="Ask me anything and press Enter…",
                 container=False,
-                scale=7,
             )
-            
-            # Add a file download component
-            download_btn = gr.Button("Download File", visible=False)
-            file_output = gr.File(label="Generated Document", visible=False)
-            
-            # Store the latest file path
+
+            download_btn = gr.Button("⬇️ Download File", visible=False)
+            file_output = gr.File(label="Generated File", visible=False)
+
             self._latest_file_path = None
-            
+
             def respond(message, chat_history):
                 agent_response = self.agent.run(message)
                 chat_history.append((message, agent_response))
-                
-                # Check if response contains a file path
-                import re
-                file_paths = re.findall(r'File ready for download: .+ \((application/[\w.+-]+)\)', agent_response)
-                
-                show_download = False
+
                 self._latest_file_path = None
-                
-                # Look for generated file paths in the response
-                paths = re.findall(r'/tmp/\w+/generated_document\.(docx|pdf)', agent_response)
-                if paths:
-                    self._latest_file_path = paths[0]
+                show_download = False
+
+                # ✅ Detect ANY generated file (docx, pdf, pptx)
+                match = re.search(
+                    r"(/tmp/.+?/generated_[\w-]+\.(docx|pdf|pptx))",
+                    agent_response,
+                )
+
+                if match:
+                    self._latest_file_path = match.group(1)
                     show_download = True
-                
-                return chat_history, gr.Button.update(visible=show_download), gr.File.update(visible=False)
-            
+
+                return (
+                    chat_history,
+                    gr.Button.update(visible=show_download),
+                    gr.File.update(visible=False),
+                )
+
             def prepare_download():
-                if self._latest_file_path:
-                    return gr.File.update(value=self._latest_file_path, visible=True)
+                if self._latest_file_path and os.path.exists(self._latest_file_path):
+                    return gr.File.update(
+                        value=self._latest_file_path,
+                        visible=True,
+                    )
                 return gr.File.update(visible=False)
-            
-            # Connect the components
+
             msg.submit(respond, [msg, chatbot], [chatbot, download_btn, file_output])
             download_btn.click(prepare_download, [], [file_output])
-            
-            gr.Markdown("Powered by smolagents and Qwen")
-            
+
+            gr.Markdown("Powered by smolagents + Qwen")
+
         return interface
 
-CustomGradioUI(agent).launch()
+
+# =========================
+# 🚀 ENTRY POINT (CRITICAL)
+# =========================
+if __name__ == "__main__":
+    CustomGradioUI(agent).launch()