Update app.py

app.py

@@ -1586,7 +1586,7 @@ client = OpenAI(api_key= os.getenv('OPENAI_API_KEY'), organization=os.getenv('OP
 #MODEL = "gpt-4o"
 MODEL = "gpt-4o-2024-05-13"
 
-def process_text():
+def process_text_old():
     text_input = st.text_input("Enter your text:")
     if text_input:
         completion = client.chat.completions.create(
@@ -1598,6 +1598,37 @@ def process_text():
         )
         st.write("Assistant: " + completion.choices[0].message.content)
 
+
+
+def process_text():
+    text_input = st.text_input("Enter your text:")
+    if text_input:
+    
+        st.session_state.messages.append({"role": "user", "content": text_input})
+        with st.chat_message("user"):
+            st.markdown(text_input)
+        with st.chat_message("assistant"):
+            completion = client.chat.completions.create(
+                model=MODEL,
+                messages=[
+                    {"role": m["role"], "content": m["content"]}
+                    for m in st.session_state.messages
+                ],
+                stream=True
+            )
+            return_text = completion.choices[0].message.content
+            st.write("Assistant: " + return_text)
+            filename = generate_filename(text_input, "md")
+            create_file(filename, text_input, return_text, should_save)
+            #return return_text
+            st.session_state.messages.append({"role": "assistant", "content": response})
+
+        #st.write("Assistant: " + completion.choices[0].message.content)
+
+
+
+
+
 def save_image(image_input, filename):
     # Save the uploaded image file
     with open(filename, "wb") as f:
@@ -1807,10 +1838,7 @@ def process_text2(MODEL='gpt-4o-2024-05-13', text_input='What is 2+2 and what is
     if text_input:
         completion = client.chat.completions.create(
             model=MODEL,
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant. Help me with my math homework!"},
-                {"role": "user", "content": f"Hello! Could you solve {text_input}?"}
-            ]
+            messages=st.session_state.messages
         )
         return_text = completion.choices[0].message.content
         st.write("Assistant: " + return_text)
@@ -1829,21 +1857,27 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
 if st.button("Clear Session"):
     st.session_state.messages = []
+
+current_messages=[]
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
+        current_messages.append(message)
         st.markdown(message["content"])
 
-if prompt := st.chat_input("GPT-4o Multimodal Chat - What can I help you with?"):
+if prompt := st.chat_input("GPT-4o Multimodal ChatBot - What can I help you with?"):
     st.session_state.messages.append({"role": "user", "content": prompt})
+
     with st.chat_message("user"):
         st.markdown(prompt)
+    
     with st.chat_message("assistant"):
         completion = client.chat.completions.create(
             model=MODEL,
-            messages=[
-                {"role": m["role"], "content": m["content"]}
-                for m in st.session_state.messages
-            ],
+            #messages=[
+            #    {"role": m["role"], "content": m["content"]}
+            #    for m in st.session_state.messages
+            #],
+            messages = st.session_state.messages,
             stream=True
         )
         response = process_text2(text_input=prompt)