Update app.py

app.py

@@ -22,14 +22,17 @@ client = OpenAI(
 
 def respond(
     message,
-    history: list[tuple[str, str]],
+    history,
     system_message,
     max_tokens,
     temperature,
     top_p,
 ):
-    messages = [{"role": "system", "content": system_message}]
+    if history is None:
+        history = []
 
+    messages = [{"role": "system", "content": system_message}]
+    
     for val in history:
         if val[0]:
             messages.append({"role": "user", "content": val[0]})
@@ -52,9 +55,15 @@ def respond(
         
         response += token
         yield response
+    
+    history.append((message, response))
+    return history
 
-def save_as_file(input_text, output_text, conversion_type):
+def save_as_file(history, conversion_type):
+    input_text = "\n".join([f"User: {h[0]}" for h in history if h[0]])
+    output_text = "\n".join([f"Assistant: {h[1]}" for h in history if h[1]])
     file_name = None
+    
     if conversion_type == "PDF":
         pdf = FPDF()
         pdf.add_page()
@@ -78,36 +87,36 @@ def save_as_file(input_text, output_text, conversion_type):
 def convert_and_download(history, conversion_type):
     if not history:
         return None
-    
-    input_text = "\n".join([f"User: {h[0]}" for h in history if h[0]])
-    output_text = "\n".join([f"Assistant: {h[1]}" for h in history if h[1]])
 
-    file_path = save_as_file(input_text, output_text, conversion_type)
+    file_path = save_as_file(history, conversion_type)
     return file_path
 
 demo = gr.Blocks(css=css)
 
 with demo:
+    history_state = gr.State([])  # Initialize an empty list to store the conversation history
+
     with gr.Row():
         system_message = gr.Textbox(value="", label="System message")
         max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
         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")
+        top_p = gr.Slider(minimum=0.1, maximum 1.0, value=0.95, step=0.05, label="Top-P")
     
     chatbot = gr.ChatInterface(
         fn=respond,
         additional_inputs=[system_message, max_tokens, temperature, top_p],
+        state=history_state,  # Pass the history state to the ChatInterface
     )
     
     with gr.Row():
-        conversion_type = gr.Dropdown(choices=["PDF", "DOCX", "TXT"], label="Conversion Type")
+        conversion_type = gr.Dropdown(choices=["PDF", "DOCX", "TXT"], value="PDF", label="Conversion Type")
         download_button = gr.Button("Convert and Download")
     
     file_output = gr.File()
 
     download_button.click(
         convert_and_download,
-        inputs=[chatbot.history, conversion_type],
+        inputs=[history_state, conversion_type],  # Pass the history state to the conversion function
         outputs=file_output
     )