Update app.py

app.py

@@ -3,6 +3,7 @@ from openai import OpenAI
 import os
 from fpdf import FPDF  # For PDF conversion
 from docx import Document  # For DOCX conversion
+import tempfile
 
 css = '''
 .gradio-container{max-width: 1000px !important}
@@ -53,26 +54,25 @@ def respond(
         yield response
 
 def save_as_file(input_text, output_text, conversion_type):
+    file_name = None
     if conversion_type == "PDF":
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", size=12)
         pdf.multi_cell(0, 10, f"User Query: {input_text}\n\nResponse: {output_text}")
-        file_name = "output.pdf"
+        file_name = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
         pdf.output(file_name)
     elif conversion_type == "DOCX":
         doc = Document()
         doc.add_heading('Conversation', 0)
         doc.add_paragraph(f"User Query: {input_text}\n\nResponse: {output_text}")
-        file_name = "output.docx"
+        file_name = tempfile.NamedTemporaryFile(delete=False, suffix=".docx").name
         doc.save(file_name)
     elif conversion_type == "TXT":
-        file_name = "output.txt"
+        file_name = tempfile.NamedTemporaryFile(delete=False, suffix=".txt").name
         with open(file_name, "w") as f:
             f.write(f"User Query: {input_text}\n\nResponse: {output_text}")
-    else:
-        return None
-
+    
     return file_name
 
 def convert_and_download(history, conversion_type):
@@ -85,10 +85,6 @@ def convert_and_download(history, conversion_type):
     file_path = save_as_file(input_text, output_text, conversion_type)
     return file_path
 
-def handle_conversion(history, conversion_type):
-    file_path = convert_and_download(history, conversion_type)
-    return gr.File(file_path)
-
 demo = gr.Blocks(css=css)
 
 with demo:
@@ -98,22 +94,20 @@ with demo:
         temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
         top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
     
-    history = gr.State(value=[])
-    
     chatbot = gr.ChatInterface(
         fn=respond,
         additional_inputs=[system_message, max_tokens, temperature, top_p],
     )
-
+    
     with gr.Row():
         conversion_type = gr.Dropdown(choices=["PDF", "DOCX", "TXT"], label="Conversion Type")
         download_button = gr.Button("Convert and Download")
     
     file_output = gr.File()
-    
+
     download_button.click(
-        handle_conversion,
-        inputs=[history, conversion_type],
+        convert_and_download,
+        inputs=[chatbot.history, conversion_type],
         outputs=file_output
     )