fix error, add history reset button

app.py

@@ -13,26 +13,28 @@ client = OpenAI()
 
 def chat_with_gpt(user_input, system_message, temperature, history):
     if not history:
-        # If starting a new conversation, add the system message first
         history = [{"role": "system", "content": system_message}]
-
-    # Append the latest user message
+    
     history.append({"role": "user", "content": user_input})
 
-    # Get response from GPT-3.5 Turbo
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",
         messages=history,
         temperature=temperature
     )
 
-    # Correctly accessing the assistant's response message content
-    assistant_message = response.choices[0].message.content  # Updated to correctly access the content attribute
+    assistant_message = response.choices[0].message.content
     history.append({"role": "assistant", "content": assistant_message})
-        
-    return history, history  # Return updated history for both display and state
 
-# Gradio interface
+    formatted_history = [(msg["content"], msg["role"]) for msg in history]
+    return formatted_history, history
+
+def reset_history(system_message):
+    # Start a new conversation with only the system message
+    new_history = [{"role": "system", "content": system_message}]
+    formatted_history = [(msg["content"], msg["role"]) for msg in new_history]
+    return formatted_history, new_history
+
 with gr.Blocks() as demo:
     gr.Markdown("### Chatte mit deinem Promptverteidiger")
     with gr.Row():
@@ -40,12 +42,11 @@ with gr.Blocks() as demo:
         user_input = gr.Textbox(label="Deine Nachricht", placeholder="Gib hier deine Chatnachricht ein...")
         temperature_slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label="Temperatur")
         submit_button = gr.Button("Senden")
+        reset_button = gr.Button("Chatverlauf zurücksetzen")
     chat_container = gr.Chatbot(label="Chatverlauf")
-    history_state = gr.State([])  # Using Gradio State to maintain conversation history
+    history_state = gr.State([])
 
-    submit_button.click(fn=chat_with_gpt, 
-                        inputs=[user_input, system_message, temperature_slider, history_state], 
-                        outputs=[chat_container, history_state])
+    submit_button.click(fn=chat_with_gpt, inputs=[user_input, system_message, temperature_slider, history_state], outputs=[chat_container, history_state])
+    reset_button.click(fn=reset_history, inputs=system_message, outputs=[chat_container, history_state])  # Reset click action
 
-# Launch the Gradio app
 demo.launch()