Update app.py

app.py

@@ -100,42 +100,6 @@ with st.sidebar:
 # Main chat area
 st.title("Personalized Real-Time Chat")
 
-# Display chat messages
-chat_container = st.container()
-
-# Input for new message
-new_message = st.text_input("Type your message:")
-if st.button("Send"):
-    if new_message:
-        timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
-        st.session_state.messages.append({
-            'user': st.session_state.current_user['name'],
-            'message': new_message,
-            'timestamp': timestamp
-        })
-        save_data()
-        st.experimental_rerun()
-
-# Function to display chat messages
-def display_messages():
-    for msg in st.session_state.messages:
-        with chat_container.container():
-            st.write(f"**{msg['user']}** ({msg['timestamp']}): {msg['message']}")
-
-# Display messages
-display_messages()
-
-# Polling for updates
-if st.button("Refresh Chat"):
-    load_data()
-    st.experimental_rerun()
-
-# Auto-refresh (note: this will refresh the entire app)
-time.sleep(5)
-st.experimental_rerun()
-
-# Additional functionalities for text, image, audio, and video processing
-
 # Function to generate filenames
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
@@ -144,29 +108,36 @@ def generate_filename(prompt, file_type):
     safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
+# Function to create files
+def create_file(filename, prompt, response, user_name, timestamp, is_image=False):
+    with open(filename, "w", encoding="utf-8") as f:
+        f.write(f"User: {user_name}\nTimestamp: {timestamp}\n\nPrompt:\n{prompt}\n\nResponse:\n{response}")
+
 # Function to process text
-def process_text(text_input):
+def process_text(user_name, text_input):
     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"):
+        timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
+        st.session_state.messages.append({"user": user_name, "message": text_input, "timestamp": timestamp})
+        with st.chat_message(user_name):
+            st.markdown(f"{user_name} ({timestamp}): {text_input}")
+        with st.chat_message("Assistant"):
             completion = client.chat.completions.create(
                 model=MODEL,
                 messages=[
-                    {"role": m["role"], "content": m["content"]}
+                    {"role": "user", "content": m["message"]}
                     for m in st.session_state.messages
                 ],
                 stream=False
             )
             return_text = completion.choices[0].message.content
-            st.write("Assistant: " + return_text)
+            st.markdown(f"Assistant ({timestamp}): {return_text}")
             filename = generate_filename(text_input, "md")
-            create_file(filename, text_input, return_text, should_save)
-            st.session_state.messages.append({"role": "assistant", "content": return_text})
+            create_file(filename, text_input, return_text, user_name, timestamp)
+            st.session_state.messages.append({"user": "Assistant", "message": return_text, "timestamp": timestamp})
+            save_data()
 
 # Function to process image
-def process_image(image_input, user_prompt):
+def process_image(user_name, image_input, user_prompt):
     if isinstance(image_input, str):
         with open(image_input, "rb") as image_file:
             image_input = image_file.read()
@@ -184,28 +155,33 @@ def process_image(image_input, user_prompt):
     )
     image_response = response.choices[0].message.content
     st.markdown(image_response)
+    timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
     filename_md = generate_filename(user_prompt, "md")
-    create_file(filename_md, image_response, '', True)
+    create_file(filename_md, user_prompt, image_response, user_name, timestamp)
+    st.session_state.messages.append({"user": user_name, "message": image_response, "timestamp": timestamp})
+    save_data()
     return image_response
 
 # Function to process audio
-def process_audio(audio_input, text_input):
+def process_audio(user_name, audio_input, text_input):
     if audio_input:
         transcription = client.audio.transcriptions.create(
             model="whisper-1",
             file=audio_input,
         )
-        st.session_state.messages.append({"role": "user", "content": transcription.text})
-        with st.chat_message("assistant"):
+        timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
+        st.session_state.messages.append({"user": user_name, "message": transcription.text, "timestamp": timestamp})
+        with st.chat_message(user_name):
+            st.markdown(f"{user_name} ({timestamp}): {transcription.text}")
+        with st.chat_message("Assistant"):
             st.markdown(transcription.text)
-            SpeechSynthesis(transcription.text)
             filename = generate_filename(transcription.text, "wav")
-            create_audio_file(filename, audio_input, should_save)
-        filename = generate_filename(transcription.text, "md")
-        create_file(filename, transcription.text, transcription.text, should_save)
+            create_file(filename, text_input, transcription.text, user_name, timestamp)
+            st.session_state.messages.append({"user": "Assistant", "message": transcription.text, "timestamp": timestamp})
+            save_data()
 
 # Function to process video
-def process_video(video_input, user_prompt):
+def process_video(user_name, video_input, user_prompt):
     if isinstance(video_input, str):
         with open(video_input, "rb") as video_file:
             video_input = video_file.read()
@@ -226,8 +202,11 @@ def process_video(video_input, user_prompt):
     )
     video_response = response.choices[0].message.content
     st.markdown(video_response)
+    timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
     filename_md = generate_filename(user_prompt, "md")
-    create_file(filename_md, video_response, '', True)
+    create_file(filename_md, user_prompt, video_response, user_name, timestamp)
+    st.session_state.messages.append({"user": user_name, "message": video_response, "timestamp": timestamp})
+    save_data()
     return video_response
 
 # Function to extract video frames
@@ -260,11 +239,6 @@ def process_audio_for_video(video_input):
     except:
         return ''
 
-# Function to create files
-def create_file(filename, prompt, response, is_image=False):
-    with open(filename, "w", encoding="utf-8") as f:
-        f.write(prompt + "\n\n" + response)
-
 # Initialize OpenAI client
 openai.api_key = os.getenv('OPENAI_API_KEY')
 openai.organization = os.getenv('OPENAI_ORG_ID')
@@ -272,6 +246,15 @@ client = OpenAI(api_key=openai.api_key, organization=openai.organization)
 MODEL = "gpt-4o-2024-05-13"
 should_save = st.sidebar.checkbox("💾 Save", value=True, help="Save your session data.")
 
+# Function to display chat messages
+def display_messages():
+    for msg in st.session_state.messages:
+        with st.chat_message(msg['user']):
+            st.markdown(f"**{msg['user']}** ({msg['timestamp']}): {msg['message']}")
+
+# Display messages
+display_messages()
+
 # Main function
 def main():
     st.markdown("##### GPT-4o Omni Model: Text, Audio, Image, & Video")
@@ -279,21 +262,36 @@ def main():
     if option == "Text":
         text_input = st.text_input("Enter your text:")
         if text_input:
-            process_text(text_input)
+            process_text(st.session_state.current_user['name'], text_input)
     elif option == "Image":
         text_input = st.text_input("Enter text prompt to use with Image context:")
         image_input = st.file_uploader("Upload an image", type=["png"])
         if image_input:
-            process_image(image_input, text_input)
+            process_image(st.session_state.current_user['name'], image_input, text_input)
     elif option == "Audio":
         text_input = st.text_input("Enter text prompt to use with Audio context:")
         uploaded_files = st.file_uploader("Upload an audio file", type=["mp3", "wav"], accept_multiple_files=True)
         for audio_input in uploaded_files:
-            process_audio(audio_input, text_input)
+            process_audio(st.session_state.current_user['name'], audio_input, text_input)
     elif option == "Video":
         video_input = st.file_uploader("Upload a video file", type=["mp4"])
         text_input = st.text_input("Enter text prompt to use with Video context:")
         if video_input and text_input:
-            process_video(video_input, text_input)
+            process_video(st.session_state.current_user['name'], video_input, text_input)
+
+# Add buttons for quick access to different UI options
+with st.sidebar:
+    if st.button("📝 Add Text"):
+        st.session_state.ui_option = "Text"
+    if st.button("🖼️ Add Image"):
+        st.session_state.ui_option = "Image"
+    if st.button("🎵 Add Audio"):
+        st.session_state.ui_option = "Audio"
+    if st.button("🎥 Add Video"):
+        st.session_state.ui_option = "Video"
 
-main()
+if 'ui_option' in st.session_state:
+    main()
+else:
+    st.session_state.ui_option = "Text"
+    main()