Update app.py

app.py

@@ -161,65 +161,10 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Custom Gradio UI with file download capability
-class CustomGradioUI(GradioUI):
-    def build_interface(self):
-        with gr.Blocks() as interface:
-            with gr.Row():
-                with gr.Column(scale=1):
-                    gr.Markdown("# AI Assistant")
-                
-            chatbot = gr.Chatbot(height=600)
-            msg = gr.Textbox(
-                placeholder="Ask me anything...",
-                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
-            self._latest_file_path = None # Instance variable to store path between calls
-            
-            def respond(message, chat_history):
-                agent_response = self.agent.run(message)
-                chat_history.append((message, agent_response))
-                
-                show_download = False
-                # Reset latest_file_path to ensure it's fresh for each response
-                # self._latest_file_path = None # Actually, should be set if file is found
 
-                # Look for generated file paths in the agent's response.
-                # tempfile.mkdtemp() usually creates paths like /tmp/some_random_string/
-                # The tool consistently names files "generated_document.ext"
-                paths = re.findall(r'/tmp/[^/]+/generated_document\.(docx|pdf|txt)', agent_response)
-                if paths:
-                    self._latest_file_path = paths[0] # Store the full path
-                    show_download = True
-                else:
-                    self._latest_file_path = None # Clear if no path found in this response
-                
-                return chat_history, gr.Button.update(visible=show_download), gr.File.update(visible=False) # Keep file output hidden until download clicked
-            
-            def prepare_download():
-                if self._latest_file_path and os.path.exists(self._latest_file_path):
-                    return gr.File.update(value=self._latest_file_path, visible=True)
-                # Handle case where file might have been cleaned up or path is stale
-                gr.Warning("File not available for download. It might have been temporary or an error occurred.")
-                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") # Qwen is mentioned, perhaps the HF endpoint uses it.
-            
-        return interface
 
 # Ensure this conditional execution if the script can also be imported
 if __name__ == "__main__":
     # Note: The original script had GradioUI(agent).launch() directly.
     # Using the custom UI:
-    CustomGradioUI(agent).launch()
+    GradioUI(agent).launch()