Add examples

app.py

@@ -31,7 +31,7 @@ def parse_news_item(html: str) -> dict:
 
 
 # %%
-def search_news(keyword, page=1):
+def search_news(keyword, page=1) -> list:
     """
     Fetch news articles related to a keyword from udn.com.
     
@@ -40,7 +40,7 @@ def search_news(keyword, page=1):
         page: The page number to fetch (default is 1).
     
     Returns:
-        A list of dictionaries containing news article data.
+        A list of dictionaries containing link, time, headline and text of news article data.
     """
     url = f"https://money.udn.com/search/result/1001/{keyword}/{page}"
     response = requests.get(url)  
@@ -66,7 +66,7 @@ def search_news(keyword, page=1):
 
 # %%
 # write a function to get the url and parse the content
-def get_content(url):
+def get_content(url) -> dict:
     """
     Fetch and parse the content of a given URL.
     
@@ -117,13 +117,13 @@ def newsAgent(task: str) -> str:
     """
     result = ""
     with ToolCollection.from_mcp(server_parameters, trust_remote_code=True) as mcp_tools:
-        agent = CodeAgent(tools=[*mcp_tools.tools], model=model)
+        agent = CodeAgent(tools=[*mcp_tools.tools[:2]], model=model)
         for event in agent.run(task, stream=True, max_steps=5):
             if isinstance(event, ActionStep):
-                result = f"\n======Step {event.step_number}======\n{event.code_action}"
+                result += f"\n## ======Step {event.step_number}======\n### Action\n```python\n{event.code_action}\n```\n### Observation\n{event.observations}"
                 # yield result
             if isinstance(event, FinalAnswerStep):
-                result = f"\n======Final======\n{event.output}"
+                result += f"\n## ======Final======\n{event.output}"
                 # yield result
     return result
 
@@ -142,19 +142,52 @@ def main():
             page = gr.Number(label="Page Number", value=1, step=1)
             search_button = gr.Button("Search")
             search_results = gr.DataFrame(label="Search Results", headers=["Link", "Time", "Headline", "Text"])
-            
+            # Examples for Search News tab
+            gr.Examples(
+                examples=[
+                    ["AI", 1],
+                    ["華碩", 2]
+                ],
+                inputs=[keyword, page],
+                outputs=search_results,
+                fn=search_news,
+                cache_examples=False
+            )
             search_button.click(search_news, inputs=[keyword, page], outputs=search_results)
+            
         
         with gr.Tab("Get Content from URL"):
             url_input = gr.Textbox(label="URL", placeholder="Enter URL to fetch content")
             content_output = gr.JSON(label="Content Output")
+            # Examples for Get Content of News tab
+            gr.Examples(
+                examples=[
+                    ["https://money.udn.com/money/story/5722/8870335?from=edn_search_result"],
+                    ["https://money.udn.com/money/story/5612/8868152?from=edn_search_result"]
+                ],
+                inputs=[url_input],
+                outputs=content_output,
+                fn=get_content,
+                cache_examples=False
+            )
             url_input.submit(get_content, inputs=url_input, outputs=content_output)
 
         with gr.Tab("News Agent"):
             agent_input = gr.Textbox(label="Task", placeholder="Enter the task")
-            run_button = gr.Button("Run")
-            result_output = gr.Textbox(label="Result", lines=10)
-            run_button.click(newsAgent, inputs=agent_input, outputs=result_output)
+            # run_button = gr.Button("Run")
+            result_output = gr.Markdown(label="Result")
+             # Examples for Get Content of News tab
+            gr.Examples(
+                examples=[
+                    ["華碩今日新聞"],
+                    ["華碩和Nvidia今日新聞"]
+                ],
+                inputs=[agent_input],
+                outputs=result_output,
+                fn=newsAgent,
+                cache_examples=True
+            )
+            agent_input.submit(newsAgent, inputs=agent_input, outputs=result_output)
 
     demo.launch(mcp_server=True, server_name="0.0.0.0",allowed_paths=["/"], share=True)