Update app.py

app.py

@@ -307,6 +307,45 @@ def process_video(user_name, video_input, user_prompt):
     save_data()
     return video_response
 
+# Callback function for text processing
+def process_text_callback(column_name):
+    text_input = st.session_state[f"{column_name}_text"]
+    selected_model = st.session_state[f"{column_name}_model"]
+    temp_values = st.session_state[f"{column_name}_temp"]
+    if text_input:
+        process_text(st.session_state.current_user['name'], text_input, selected_model, temp_values)
+        st.session_state[f"{column_name}_text"] = ""  # Clear the input after processing
+
+# Callback function for image processing
+def process_image_callback(column_name):
+    text_input = st.session_state[f"{column_name}_image_text"]
+    uploaded_files = st.session_state[f"{column_name}_image_upload"]
+    if text_input and uploaded_files:
+        for image_input in uploaded_files:
+            image_bytes = image_input.read()
+            process_image(st.session_state.current_user['name'], image_bytes, text_input)
+        st.session_state[f"{column_name}_image_text"] = ""  # Clear the input after processing
+        st.session_state[f"{column_name}_image_upload"] = None  # Clear the file uploader
+
+# Callback function for audio processing
+def process_audio_callback(column_name):
+    text_input = st.session_state[f"{column_name}_audio_text"]
+    uploaded_files = st.session_state[f"{column_name}_audio_upload"]
+    if uploaded_files:
+        for audio_input in uploaded_files:
+            process_audio(st.session_state.current_user['name'], audio_input, text_input)
+        st.session_state[f"{column_name}_audio_text"] = ""  # Clear the input after processing
+        st.session_state[f"{column_name}_audio_upload"] = None  # Clear the file uploader
+
+# Callback function for video processing
+def process_video_callback(column_name):
+    text_input = st.session_state[f"{column_name}_video_text"]
+    video_input = st.session_state[f"{column_name}_video_upload"]
+    if video_input and text_input:
+        process_video(st.session_state.current_user['name'], video_input, text_input)
+        st.session_state[f"{column_name}_video_text"] = ""  # Clear the input after processing
+        st.session_state[f"{column_name}_video_upload"] = None  # Clear the file uploader
+
 # Main function for each column
 def main_column(column_name):
     st.markdown(f"##### {column_name}")
@@ -316,40 +355,20 @@ def main_column(column_name):
     option = st.selectbox(f"Select an option for {column_name}", ("Text", "Image", "Audio", "Video"), key=f"{column_name}_option")
     
     if option == "Text":
-        text_input = st.text_input(f"Enter your text for {column_name}:", key=f"{column_name}_text")
-        if st.button(f"Process Text for {column_name}"):
-            if text_input:
-                process_text(st.session_state.current_user['name'], text_input, selected_model, temp_values)
-                # Clear the input after processing
-                st.session_state[f"{column_name}_text"] = ""
+        st.text_input(f"Enter your text for {column_name}:", key=f"{column_name}_text")
+        st.button(f"Process Text for {column_name}", on_click=process_text_callback, args=(column_name,))
     elif option == "Image":
-        text_input = st.text_input(f"Enter text prompt to use with Image context for {column_name}:", key=f"{column_name}_image_text")
-        uploaded_files = st.file_uploader(f"Upload images for {column_name}", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key=f"{column_name}_image_upload")
-        if st.button(f"Process Image for {column_name}"):
-            for image_input in uploaded_files:
-                image_bytes = image_input.read()
-                process_image(st.session_state.current_user['name'], image_bytes, text_input)
-            # Clear the inputs after processing
-            st.session_state[f"{column_name}_image_text"] = ""
-            st.session_state[f"{column_name}_image_upload"] = None
+        st.text_input(f"Enter text prompt to use with Image context for {column_name}:", key=f"{column_name}_image_text")
+        st.file_uploader(f"Upload images for {column_name}", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key=f"{column_name}_image_upload")
+        st.button(f"Process Image for {column_name}", on_click=process_image_callback, args=(column_name,))
     elif option == "Audio":
-        text_input = st.text_input(f"Enter text prompt to use with Audio context for {column_name}:", key=f"{column_name}_audio_text")
-        uploaded_files = st.file_uploader(f"Upload an audio file for {column_name}", type=["mp3", "wav"], accept_multiple_files=True, key=f"{column_name}_audio_upload")
-        if st.button(f"Process Audio for {column_name}"):
-            for audio_input in uploaded_files:
-                process_audio(st.session_state.current_user['name'], audio_input, text_input)
-            # Clear the inputs after processing
-            st.session_state[f"{column_name}_audio_text"] = ""
-            st.session_state[f"{column_name}_audio_upload"] = None
+        st.text_input(f"Enter text prompt to use with Audio context for {column_name}:", key=f"{column_name}_audio_text")
+        st.file_uploader(f"Upload an audio file for {column_name}", type=["mp3", "wav"], accept_multiple_files=True, key=f"{column_name}_audio_upload")
+        st.button(f"Process Audio for {column_name}", on_click=process_audio_callback, args=(column_name,))
     elif option == "Video":
-        video_input = st.file_uploader(f"Upload a video file for {column_name}", type=["mp4"], key=f"{column_name}_video_upload")
-        text_input = st.text_input(f"Enter text prompt to use with Video context for {column_name}:", key=f"{column_name}_video_text")
-        if st.button(f"Process Video for {column_name}"):
-            if video_input and text_input:
-                process_video(st.session_state.current_user['name'], video_input, text_input)
-            # Clear the inputs after processing
-            st.session_state[f"{column_name}_video_text"] = ""
-            st.session_state[f"{column_name}_video_upload"] = None
+        st.file_uploader(f"Upload a video file for {column_name}", type=["mp4"], key=f"{column_name}_video_upload")
+        st.text_input(f"Enter text prompt to use with Video context for {column_name}:", key=f"{column_name}_video_text")
+        st.button(f"Process Video for {column_name}", on_click=process_video_callback, args=(column_name,))
 
 # Main Streamlit app
 st.title("Personalized Real-Time Chat")
@@ -390,4 +409,4 @@ with col2:
 
 # Run the Streamlit app
 if __name__ == "__main__":
-    st.markdown("\n[by Aaron Wacker](https://huggingface.co/spaces/awacke1/).")
+    st.markdown("\n[by Aaron Wacker](https://huggingface.co/spaces/awacke1/ChatStreamlitMultiplayer).")