Update app.py

app.py

@@ -1381,11 +1381,11 @@ if filename is not None:
 session_state = {}
 if "search_queries" not in session_state:
     session_state["search_queries"] = []
+
 example_input = st.text_input("AI Search ArXiV Scholarly Articles", value=session_state["search_queries"][-1] if session_state["search_queries"] else "")
+
 if example_input:
     session_state["search_queries"].append(example_input)
-
-    # Search AI
     query=example_input
     if query: 
         result = search_arxiv(query)
@@ -1462,27 +1462,14 @@ 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_old():
-    text_input = st.text_input("Enter your text:")
-    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}?"}
-            ]
-        )
-        st.write("Assistant: " + completion.choices[0].message.content)
-
-
-
-def process_text():
-    text_input = st.text_input("Enter your text:")
+def process_text(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"):
             completion = client.chat.completions.create(
                 model=MODEL,
@@ -1490,14 +1477,13 @@ def process_text():
                     {"role": m["role"], "content": m["content"]}
                     for m in st.session_state.messages
                 ],
-                stream=True
+                stream=False
             )
             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.session_state.messages.append({"role": "assistant", "content": return_text})
 
         #st.write("Assistant: " + completion.choices[0].message.content)
 
@@ -1673,8 +1659,12 @@ def process_audio_and_video(video_input):
             ],
             temperature=0,
         )
-    
-        st.markdown(response.choices[0].message.content)
+        results = response.choices[0].message.content
+        st.markdown(results) 
+        
+        filename = generate_filename(transcript, "md")
+        create_file(filename, transcript, results, should_save)
+
 
 
 def main():
@@ -1682,7 +1672,10 @@ def main():
     st.markdown("##### GPT-4o Omni Model: Text, Audio, Image, & Video")
     option = st.selectbox("Select an option", ("Text", "Image", "Audio", "Video"))
     if option == "Text":
-        process_text()
+        text_input = st.text_input("Enter your text:")
+        if (text_input > ''):
+            textResponse = process_text(text_input)
+
     elif option == "Image":
         image_input = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
         image_response = process_image(image_input)