Update app.py

app.py

@@ -27,6 +27,10 @@ def handle_and_display_image(file_id):
         print('Failed to download file.')
         return None
 
+
+if "api_call_status" not in st.session_state:
+    st.session_state.api_call_status = None
+
 # Initialize global session state
 if "session_id" not in st.session_state:
     st.session_state.session_id = str(uuid.uuid4())
@@ -83,63 +87,115 @@ def page2():
 
 def page1():
     st.sidebar.title("OpenAI - Assistant API🤖")
-    # ... [other sidebar setup] ...
+    st.sidebar.markdown("**Model:** gpt-4-1106-preview API")
+    st.sidebar.markdown("**Version:** 1.1")
+    st.sidebar.markdown("**Session-ID:**")
+    st.sidebar.markdown(st.session_state.session_id)
+    st.sidebar.divider()
+    st.title("Assitent-Name")
+    st.text("Beschreibung")
+
 
-    # Initialize or retrieve the assistant, thread, and other states
-    if "assistant" not in st.session_state:
+   if "assistant" not in st.session_state:
         openai.api_key = api_key
         st.session_state.assistant = openai.beta.assistants.retrieve(assistant_id)
         st.session_state.thread = client.beta.threads.create(metadata={'session_id': st.session_state.session_id})
 
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
-
-    if "retry_error" not in st.session_state:
-        st.session_state.retry_error = 0
-
-    # User inputs a query
+    # User input
     prompt = st.text_input("Wie kann ich dir helfen?")
 
+    # Submit button
     if st.button("Submit"):
-        try:
-            # Make an API call here with the prompt
-            response = openai.Completion.create(
-                model=MODEL,  # Specify the model, e.g., "text-davinci-003"
-                prompt=prompt,
-                max_tokens=150  # Adjust as needed
-            )
-            st.session_state.messages.append(("User", prompt))
-            st.session_state.messages.append(("Assistant", response.choices[0].text.strip()))
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
+        st.session_state.api_call_status = "pending"
+        # Here, use the prompt to make your API call
+        # For example, adding the prompt to the thread and creating a run
+        response = "This is a simulated response"  # Replace with actual response
+        st.session_state.messages.append(("User", prompt))
+        st.session_state.messages.append(("Assistant", response))
+        st.session_state.api_call_status = "completed"
+
+
+    # Display messages
+    for role, message in st.session_state.messages:
+        st.text(f"{role}: {message}")
+
+
+       
+    # If the run is completed, display the messages
+    elif hasattr(st.session_state.run, 'status') and st.session_state.run.status == "completed":
+        # Retrieve the list of messages
+        st.session_state.messages = client.beta.threads.messages.list(
+            thread_id=st.session_state.thread.id
+        )
+
+
+        # Display messages
+        for message in reversed(st.session_state.messages.data):
+            if message.role in ["user", "assistant"]:
+                with st.chat_message(message.role):
+                    for content_part in message.content:
+                        if content_part.type == 'text':  # For text responses
+                            message_text = content_part.text.value
+                            st.markdown(message_text)
+                        elif hasattr(content_part, 'image_file') and content_part.image_file:  
+                            image = handle_and_display_image(content_part.image_file.file_id)
+                            if image:
+                                st.image(image, caption="Generated Image")
+
+    
+
+    # Display chat history
+    for sender, message in st.session_state.chat_history:
+        st.text(f"{sender}: {message}")
+
 
-    # Check the status of the API call
+        # Add message to the thread
+        st.session_state.messages = client.beta.threads.messages.create(
+            thread_id=st.session_state.thread.id,
+            role="user",
+            content=prompt
+        )
+
+        # Do a run to process the messages in the thread
+        st.session_state.run = client.beta.threads.runs.create(
+            thread_id=st.session_state.thread.id,
+            assistant_id=st.session_state.assistant.id,
+        )
+        if st.session_state.retry_error < 3:
+            time.sleep(1) # Wait 1 second before checking run status
+            st.rerun()
+
+    # Check if 'run' object has 'status' attribute
     if hasattr(st.session_state.run, 'status'):
+        # Handle the 'running' status
         if st.session_state.run.status == "running":
-            st.text("Thinking ......")
-        elif st.session_state.run.status == "completed":
-            # Retrieve and display messages when the run is completed
-            st.session_state.messages = client.beta.threads.messages.list(
-                thread_id=st.session_state.thread.id
-            ).data
-
-            for message in st.session_state.messages:
-                st.text(f"{message.role}: {message.content.get('text', 'No text found')}")
+            with st.chat_message('assistant'):
+                st.write("Thinking ......")
+            if st.session_state.retry_error < 3:
+                time.sleep(1)  # Short delay to prevent immediate rerun
+                st.rerun()
+
+        # Handle the 'failed' status
         elif st.session_state.run.status == "failed":
             st.session_state.retry_error += 1
+            with st.chat_message('assistant'):
+                if st.session_state.retry_error < 3:
+                    st.write("Run failed, retrying ......")
+                    time.sleep(3)  # Longer delay before retrying
+                    st.rerun()
+                else:
+                    st.error("FAILED: The OpenAI API is currently processing too many requests. Please try again later ......")
+
+        # Handle any status that is not 'completed'
+        elif st.session_state.run.status != "completed":
+            # Attempt to retrieve the run again
+            st.session_state.run = client.beta.threads.runs.retrieve(
+                thread_id=st.session_state.thread.id,
+                run_id=st.session_state.run.id,
+            )
             if st.session_state.retry_error < 3:
-                st.text("Run failed, retrying ......")
                 time.sleep(3)
                 st.rerun()
-            else:
-                st.error("FAILED: The OpenAI API is currently processing too many requests. Please try again later ......")
-        # Handle other statuses if needed
-
-    # Reset retry_error if needed
-    if st.button("Reset"):
-        st.session_state.retry_error = 0
-
 
 if __name__ == "__main__":
     main()
-