fix agent context

.gitignore

@@ -1,9 +1,97 @@
+# Python
 __pycache__/
-*.pyc
-*.pyo
-*.pyd
-*.pyw
-*.pyz
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
 
+# Poetry
 .venv/
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+pytest-report.xml
+pytest.xml
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+.DS_Store
+.history/
+
+# Logs
+logs/
+*.log
+log.txt
+
+# Database
+*.db
+*.sqlite
+*.sqlite3
+
+# Distribution / packaging
+.Python
+dist/
+build/
+*.egg-info/
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# Local development settings
+local_settings.py
+
+# Data files
+*.csv
+*.xls
+*.xlsx
+**/data/*
+!**/data/.gitkeep
+
 .gradio/

.python-version

@@ -0,0 +1 @@
+3.11

Gradio_UI.py

@@ -13,12 +13,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import mimetypes
 import os
 import re
-import shutil
 from typing import Optional
 
+import gradio as gr
 from smolagents.agent_types import (
     AgentAudio,
     AgentImage,
@@ -32,7 +31,6 @@ from smolagents.utils import _is_package_available
 
 def pull_messages_from_step(step_log: MemoryStep):
     """Extract ChatMessage objects from agent steps with proper nesting"""
-    import gradio as gr
 
     if isinstance(step_log, ActionStep):
         # Output the step number
@@ -165,7 +163,6 @@ def stream_to_gradio(
         raise ModuleNotFoundError(
             "Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
         )
-    import gradio as gr
 
     total_input_tokens = 0
     total_output_tokens = 0
@@ -225,66 +222,14 @@ class GradioUI:
                 os.mkdir(file_upload_folder)
 
     def interact_with_agent(self, prompt, messages):
-        import gradio as gr
-
         messages.append(gr.ChatMessage(role="user", content=prompt))
-        yield messages
-        for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
+        for msg in stream_to_gradio(
+            self.agent,
+            task=prompt,
+            reset_agent_memory=True,
+        ):
             messages.append(msg)
             yield messages
-        yield messages
-
-    def upload_file(
-        self,
-        file,
-        file_uploads_log,
-        allowed_file_types=[
-            "application/pdf",
-            "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-            "text/plain",
-        ],
-    ):
-        """
-        Handle file uploads, default allowed types are .pdf, .docx, and .txt
-        """
-        import gradio as gr
-
-        if file is None:
-            return gr.Textbox("No file uploaded", visible=True), file_uploads_log
-
-        try:
-            mime_type, _ = mimetypes.guess_type(file.name)
-        except Exception as e:
-            return gr.Textbox(f"Error: {e}", visible=True), file_uploads_log
-
-        if mime_type not in allowed_file_types:
-            return gr.Textbox("File type disallowed", visible=True), file_uploads_log
-
-        # Sanitize file name
-        original_name = os.path.basename(file.name)
-        sanitized_name = re.sub(
-            r"[^\w\-.]", "_", original_name
-        )  # Replace any non-alphanumeric, non-dash, or non-dot characters with underscores
-
-        type_to_ext = {}
-        for ext, t in mimetypes.types_map.items():
-            if t not in type_to_ext:
-                type_to_ext[t] = ext
-
-        # Ensure the extension correlates to the mime type
-        sanitized_name = sanitized_name.split(".")[:-1]
-        sanitized_name.append("" + type_to_ext[mime_type])
-        sanitized_name = "".join(sanitized_name)
-
-        # Save the uploaded file to the specified folder
-        file_path = os.path.join(
-            self.file_upload_folder, os.path.basename(sanitized_name)
-        )
-        shutil.copy(file.name, file_path)
-
-        return gr.Textbox(
-            f"File uploaded: {file_path}", visible=True
-        ), file_uploads_log + [file_path]
 
     def log_user_message(self, text_input, file_uploads_log):
         return (
@@ -298,8 +243,6 @@ class GradioUI:
         )
 
     def launch(self, **kwargs):
-        import gradio as gr
-
         with gr.Blocks(theme=gr.themes.Soft()) as demo:
             gr.Markdown("""
             # 🤏 📰 SmolNews: News and Time AI Agent
@@ -357,7 +300,7 @@ class GradioUI:
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
 
-        demo.launch(debug=True, share=True, **kwargs)
+        demo.launch(debug=True, share=False, **kwargs)
 
 
 __all__ = ["stream_to_gradio", "GradioUI"]

README.md

@@ -15,7 +15,9 @@ tags:
 - agent-course
 ---
 
-# News and Time Agent
+# SmolNews: News and Time Agent
+
+* Top 5 headlines, smol news but with a twist.
 
 A Gradio-based web application that combines news fetching and timezone capabilities using the smolagents framework.
 

pyproject.toml

@@ -0,0 +1,14 @@
+[project]
+name = "smolnews"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "markdownify>=1.0.0",
+    "pandas>=2.2.3",
+    "python-dateutil>=2.9.0.post0",
+    "pytz>=2025.1",
+    "requests>=2.32.3",
+    "smolagents[gradio]>=1.9.2",
+]

requirements.txt

@@ -1,7 +1,60 @@
-markdownify
-smolagents[gradio]
-requests
-duckduckgo_search
-pandas
-pytz
-python-dateutil
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.8.0
+beautifulsoup4==4.13.3
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+duckduckgo-search==7.5.0
+fastapi==0.115.11
+ffmpy==0.5.0
+filelock==3.17.0
+fsspec==2025.2.0
+gradio==5.20.0
+gradio-client==1.7.2
+groovy==0.1.2
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.29.1
+idna==3.10
+jinja2==3.1.5
+lxml==5.3.1
+markdown-it-py==3.0.0
+markdownify==1.0.0
+markupsafe==2.1.5
+mdurl==0.1.2
+numpy==2.2.3
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+primp==0.14.0
+pydantic==2.10.6
+pydantic-core==2.27.2
+pydub==0.25.1
+pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-multipart==0.0.20
+pytz==2025.1
+pyyaml==6.0.2
+requests==2.32.3
+rich==13.9.4
+ruff==0.9.9
+safehttpx==0.1.6
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+smolagents==1.9.2
+sniffio==1.3.1
+soupsieve==2.6
+starlette==0.46.0
+tomlkit==0.13.2
+tqdm==4.67.1
+typer==0.15.2
+typing-extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==15.0