Update app.py

Use openai's websearch tool. PDF download working,

app.py

@@ -18,7 +18,7 @@ from urllib.parse import quote
 from multiprocessing import context
 # from agents.tool import WebSearchTool
 from typing_extensions import TypedDict, Any
-from agents import Agent, FunctionTool, ImageGenerationTool, RunContextWrapper,  Runner, function_tool, CodeInterpreterTool, ModelSettings
+from agents import Agent, FunctionTool, ImageGenerationTool, RunContextWrapper,  Runner, WebSearchTool, function_tool, CodeInterpreterTool, ModelSettings
 from openai import OpenAI
 from markdown_pdf import MarkdownPdf
 from markdown_pdf import Section
@@ -60,7 +60,7 @@ async def make_pdf(text: str, title: str)->str:
         title:  A title that will be used as part of a filename in a file path.
 
     Returns:
-        The file path to the PDF that was created and saved.
+        The title of the PDF that was created.
     '''
     for file in glob('./document.pdf'):
         os.remove(file)
@@ -68,7 +68,7 @@ async def make_pdf(text: str, title: str)->str:
     pdf.add_section(Section(text))
     outpath = os.path.join('./','document.pdf')
     pdf.save(outpath)
-    return outpath
+    return title
 
 @function_tool
 async def get_news(query: str, window: str)->str:
@@ -272,7 +272,7 @@ async def chat(prompt_window, user_window, password, history, output_window,
     agent = Agent(name="Assistant",
                  instructions=instructions,
                  model_settings=ModelSettings(reasoning=Reasoning(effort='low', summary='detailed')),
-                 tools=[get_distance, search_web, get_news,
+                 tools=[get_distance, WebSearchTool(), get_news, make_pdf,  # search_web
             CodeInterpreterTool(tool_config={"type": "code_interpreter","container": code_container}),     # make_pdf,
               ImageGenerationTool(tool_config={"type": "image_generation", "quality": "low"},)],)
 
@@ -344,6 +344,7 @@ async def chat(prompt_window, user_window, password, history, output_window,
     exception_msg = ''
     result = None
     reasoning = ''
+    pdf_id = None
     try:
         result = await Runner.run(agent, max_turns=20,
                 input=inputs)
@@ -358,6 +359,22 @@ async def chat(prompt_window, user_window, password, history, output_window,
                     fp.write(base64.b64decode(img_result))
                 image_window = gr.Image(visible=True, value=image_out_path)
 
+            if item.type == 'tool_call_item':
+                raw = item.raw_item
+                if raw.name == 'make_pdf':
+                    pdf_id = raw.call_id
+
+            if item.type == 'tool_call_output_item' and pdf_id:
+                raw_id = item.raw_item['call_id']
+                if raw_id == pdf_id:
+                    # raw_data = item.raw_item['output']
+                    # bin_out = base64.b64decode(raw_data)
+                    # with open('./document.pdf', 'w', encoding='utf-8') as fp:
+                    #     fp.write(raw_data)
+                    file_download = gr.DownloadButton(label=f'Download PDF Doc',
+                                    visible=True, value='./document.pdf')
+                
+
             if item.type == "reasoning_item":
                 for o in item.raw_item.summary:
                     if o.type == "summary_text":
@@ -369,7 +386,11 @@ async def chat(prompt_window, user_window, password, history, output_window,
         #             for o in item.summary:
         #                 reasoning += ('\nm1: ' + o.text + '\n')
 
-        reply = md(result.final_output)
+        out_text = "\n".join(line for line in result.final_output.splitlines() if 
+                        not ('download' in line.casefold() and 'mnt' in line.casefold()))
+        if out_text == '':
+            out_text = 'Download using button below.'
+        reply = md(out_text)
         response += "\n\n***YOU***: " + prompt + "\n\n***GPT***: " +  reply.replace('```','\n\n```\n\n')
         if reasoning != '':
             response += '\n\n**Reasoning:**\n\n' + reasoning + '\n'
@@ -450,6 +471,7 @@ def show_help():
 
 def new_session(user_window, history):
     history = []
+    clean_up_files()
     return [prompt_window, history, 'Session cleared',
            gr.Image(visible=False, value=None),
            gr.Image(visible=False, value=None), [],