Update app.py

app.py

@@ -6,126 +6,123 @@ import os
 
 from openai import OpenAI
 
+openai.api_key  = os.environ['OPENAI_API_KEY']
 
-def main():
-    st.set_page_config(page_title="Manuales de Procedimientos de SCHT")
+client = OpenAI()
 
-    openai.api_key  = os.environ['OPENAI_API_KEY']
+MODEL = "gpt-4-1106-preview"
 
-    client = OpenAI()
+if "session_id" not in st.session_state: # Used to identify each session
+    st.session_state.session_id = str(uuid.uuid4())
 
-    MODEL = "gpt-4-1106-preview"
+if "run" not in st.session_state: # Stores the run state of the assistant
+    st.session_state.run = {"status": None}
 
-    if "session_id" not in st.session_state: # Used to identify each session
-        st.session_state.session_id = str(uuid.uuid4())
-    
-    if "run" not in st.session_state: # Stores the run state of the assistant
-        st.session_state.run = {"status": None}
-    
-    if "messages" not in st.session_state: # Stores the messages of the assistant
-        st.session_state.messages = []
-    
-    if "retry_error" not in st.session_state: # Used for error handling
-        st.session_state.retry_error = 0
-
-    st.sidebar.title("Manuales de Procedimientos de SCHT")
-    st.sidebar.divider()
-    st.sidebar.markdown("Versión actual: 0.0.1")
-    st.sidebar.markdown("Usando el API de gpt-4-1106-preview")
-    st.sidebar.markdown("### Incluye los siguientes manuales:")
-    st.sidebar.markdown("GAF-PROC-001 Procedimiento de Compras y Evaluacion de Proveedores")
-    st.sidebar.markdown("GAF-PROC-002 Procedimiento Reclutamiento y Seleccion de Personal")
-    st.sidebar.markdown("GAF-PROC-003 Procedimiento de Contratacion de Personal")
-    st.sidebar.markdown("GAF-PROC-004 Procedimiento Facturación y Cobranzas")
-    st.sidebar.markdown("GPD-PROC-001 Promocion y Desarrollo")
-    st.sidebar.markdown("GPR-PROC-001 Inicio del Proyecto")
-    
-    st.sidebar.markdown(st.session_state.session_id)
-    st.sidebar.divider()
-    
-    if "assistant" not in st.session_state:
-        
-        st.session_state.assistant = openai.beta.assistants.retrieve(assistant_id='asst_ZNKKXJLxzvwX7zVo0cVR1USv')
-    
-        # Create a new thread for this session
-        st.session_state.thread = client.beta.threads.create(
-            metadata={
-                'session_id': st.session_state.session_id,
-            }
-        )
+if "messages" not in st.session_state: # Stores the messages of the assistant
+    st.session_state.messages = []
+
+if "retry_error" not in st.session_state: # Used for error handling
+    st.session_state.retry_error = 0
+
+st.set_page_config(page_title="Manuales de Procedimientos de SCHT")
+st.sidebar.title("Manuales de Procedimientos de SCHT")
+st.sidebar.divider()
+st.sidebar.markdown("Versión actual: 0.0.1")
+st.sidebar.markdown("Usando el API de gpt-4-1106-preview")
+st.sidebar.markdown("### Incluye los siguientes manuales:")
+st.sidebar.markdown("GAF-PROC-001 Procedimiento de Compras y Evaluacion de Proveedores")
+st.sidebar.markdown("GAF-PROC-002 Procedimiento Reclutamiento y Seleccion de Personal")
+st.sidebar.markdown("GAF-PROC-003 Procedimiento de Contratacion de Personal")
+st.sidebar.markdown("GAF-PROC-004 Procedimiento Facturación y Cobranzas")
+st.sidebar.markdown("GPD-PROC-001 Promocion y Desarrollo")
+st.sidebar.markdown("GPR-PROC-001 Inicio del Proyecto")
+
+st.sidebar.markdown(st.session_state.session_id)
+st.sidebar.divider()
+
+if "assistant" not in st.session_state:
     
-    # Refresh Session Button
-    if st.button("Reiniciar (Se borrarán los mensajes anteriores)"):
-        st.session_state.session_id = str(uuid.uuid4())  # Reset session ID
-        st.session_state.messages = []  # Reset messages
-        st.session_state.run = {"status": None}  # Reset run state
-        st.session_state.retry_error = 0  # Reset error count
-        # You might also need to reset or recreate the assistant and thread here
+    st.session_state.assistant = openai.beta.assistants.retrieve(assistant_id='asst_ZNKKXJLxzvwX7zVo0cVR1USv')
+
+    # Create a new thread for this session
+    st.session_state.thread = client.beta.threads.create(
+        metadata={
+            'session_id': st.session_state.session_id,
+        }
+    )
+
+# Refresh Session Button
+if st.button("Reiniciar (Se borrarán los mensajes anteriores)"):
+    st.session_state.session_id = str(uuid.uuid4())  # Reset session ID
+    st.session_state.messages = []  # Reset messages
+    st.session_state.run = {"status": None}  # Reset run state
+    st.session_state.retry_error = 0  # Reset error count
+    # You might also need to reset or recreate the assistant and thread here
+    st.rerun()
+
+# 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:
+                    message_text = content_part.text.value
+                    st.markdown(message_text)
+
+if prompt := st.chat_input("Escribe tu consulta sobre los procedimientos de SCHT"):
+    with st.chat_message('user'):
+        st.write(prompt)
+
+    # 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(3) # Wait 1 second before checking run status
         st.rerun()
-    
-    # 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:
-                        message_text = content_part.text.value
-                        st.markdown(message_text)
-    
-    if prompt := st.chat_input("Escribe tu consulta sobre los procedimientos de SCHT"):
-        with st.chat_message('user'):
-            st.write(prompt)
-    
-        # 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,
-        )
+
+# 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":
+        with st.chat_message('assistant'):
+            st.write("Pensando ......")
         if st.session_state.retry_error < 3:
-            time.sleep(3) # Wait 1 second before checking run status
+            time.sleep(1)  # Short delay to prevent immediate rerun, adjust as needed
             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":
-            with st.chat_message('assistant'):
-                st.write("Pensando ......")
+
+    # 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:
-                time.sleep(1)  # Short delay to prevent immediate rerun, adjust as needed
+                st.write("Hubo un error, probando de nuevo ......")
+                time.sleep(3)  # Longer delay before retrying
                 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("Hubo un error, probando de nuevo ......")
-                    time.sleep(3)  # Longer delay before retrying
-                    st.rerun()
-                else:
-                    st.error("ERROR: La página de Open AI puede estar recibiendo demasiados pedidos, prueba en unos momentos ......")
-    
-        # Handle any status that is not 'completed'
-        elif st.session_state.run.status != "completed":
-            # Attempt to retrieve the run again, possibly redundant if there's no other status but 'running' or 'failed'
-            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:
-                time.sleep(3)
-                st.rerun()
+            else:
+                st.error("ERROR: La página de Open AI puede estar recibiendo demasiados pedidos, prueba en unos momentos ......")
+
+    # Handle any status that is not 'completed'
+    elif st.session_state.run.status != "completed":
+        # Attempt to retrieve the run again, possibly redundant if there's no other status but 'running' or 'failed'
+        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:
+            time.sleep(3)
+            st.rerun()