Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import json
 
 # Function to call the API
-def call_chatgpt_api(message):
+def call_chatgpt_api(message, dialog_messages):
     url = "https://api.safone.dev/chatgpt"
     headers = {
         "accept": "application/json",
@@ -13,7 +13,7 @@ def call_chatgpt_api(message):
         "message": message,
         "version": 3,
         "chat_mode": "assistant",
-        "dialog_messages": "[{\"bot\":\"\",\"user\":\"\"}]"
+        "dialog_messages": json.dumps(dialog_messages)
     }
 
     response = requests.post(url, headers=headers, data=json.dumps(payload))
@@ -22,21 +22,39 @@ def call_chatgpt_api(message):
     else:
         return None
 
+# Initialize session state if not already done
+if 'dialog_messages' not in st.session_state:
+    st.session_state.dialog_messages = []
+
+if 'responses' not in st.session_state:
+    st.session_state.responses = []
+
 # Streamlit App
 st.title('SlimshadowGPT')
 st.write("This is a simple interface to interact with the GPT-3.5 API.")
 
-user_input = st.text_input("Enter your message:")
+user_input = st.text_input("Enter your message:", key='user_input')
 
 if st.button('Send'):
     if user_input:
+        st.session_state.dialog_messages.append({"bot": "", "user": user_input})
+        
         with st.spinner('Generating response...'):
-            response = call_chatgpt_api(user_input)
+            response = call_chatgpt_api(user_input, st.session_state.dialog_messages)
             if response:
-                st.write("Assistant:", response['choices'][0]['message']['content'])
+                assistant_message = response['choices'][0]['message']['content']
+                st.session_state.dialog_messages.append({"bot": assistant_message, "user": ""})
+                st.session_state.responses.append((user_input, assistant_message))
             else:
                 st.write("Failed to get a response from the API.")
-    else:
-        st.write("Please enter a message to send.")
+        
+        # Clear input box after sending
+        st.session_state.user_input = ""
+
+# Display the conversation history
+st.write("### Chat History")
+for user_msg, bot_msg in st.session_state.responses:
+    st.write(f"**You:** {user_msg}")
+    st.write(f"**Assistant:** {bot_msg}")
 
 # Run the app using `streamlit run app.py` in the terminal