Update app.py

app.py

@@ -58,14 +58,16 @@ def respond(
 
 # Function to save chat history to a text file
 def save_as_txt(history):
-    with open("chat_history.txt", "w") as f:
+    file_path = "chat_history.txt"
+    with open(file_path, "w") as f:
         for user_message, assistant_message in history:
             f.write(f"User: {user_message}\n")
             f.write(f"Assistant: {assistant_message}\n")
-    return "chat_history.txt"
+    return file_path
 
 # Function to save chat history to a DOCX file
 def save_as_docx(history):
+    file_path = "chat_history.docx"
     doc = Document()
     doc.add_heading('Chat History', 0)
     
@@ -73,11 +75,12 @@ def save_as_docx(history):
         doc.add_paragraph(f"User: {user_message}")
         doc.add_paragraph(f"Assistant: {assistant_message}")
     
-    doc.save("chat_history.docx")
-    return "chat_history.docx"
+    doc.save(file_path)
+    return file_path
 
 # Function to save chat history to a PDF file
 def save_as_pdf(history):
+    file_path = "chat_history.pdf"
     buffer = BytesIO()
     c = canvas.Canvas(buffer, pagesize=letter)
     width, height = letter
@@ -99,12 +102,12 @@ def save_as_pdf(history):
     c.save()
     buffer.seek(0)
     
-    with open("chat_history.pdf", "wb") as f:
+    with open(file_path, "wb") as f:
         f.write(buffer.read())
     
-    return "chat_history.pdf"
+    return file_path
 
-# Gradio interface
+# Function to handle file saving based on format
 def handle_file_save(history, file_format):
     if file_format == "txt":
         return save_as_txt(history)
@@ -112,37 +115,15 @@ def handle_file_save(history, file_format):
         return save_as_docx(history)
     elif file_format == "pdf":
         return save_as_pdf(history)
+    return None
 
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-P",
-        ),
-        gr.Dropdown(
-            choices=["txt", "docx", "pdf"],
-            label="Save as",
-        ),
-    ],
-    outputs=[
-        gr.File(label="Download Chat History"),
-    ],
-    css=css,
-    theme="allenai/gradio-theme",
-)
-
+# Handler function for Gradio app
 def save_handler(message, history, system_message, max_tokens, temperature, top_p, file_format):
     response = respond(message, history, system_message, max_tokens, temperature, top_p)
     saved_file = handle_file_save(history, file_format)
     return saved_file
 
+# Gradio interface
 demo = gr.Interface(
     fn=save_handler,
     inputs=[