Update app.py

app.py

@@ -12,19 +12,20 @@ client = openai.OpenAI(api_key=openai.api_key)
 def ask_openai(question):
     try:
         thread = client.beta.threads.create()
+
         client.beta.threads.messages.create(
             thread_id=thread.id,
             role="user",
             content=question,
         )
-        
+
         run = client.beta.threads.runs.create(
             thread_id=thread.id,
             assistant_id=assistant_id
         )
 
         response_received = False
-        timeout = 150  
+        timeout = 30  # seconds
         start_time = time.time()
 
         while not response_received and time.time() - start_time < timeout:
@@ -35,22 +36,30 @@ def ask_openai(question):
             if run_status.status == 'completed':
                 response_received = True
             else:
-                time.sleep(4) 
+                time.sleep(4)  
 
         if not response_received:
             return "Response timed out."
 
-        messages = client.beta.threads.messages.list(
+        steps = client.beta.threads.runs.steps.list(
             thread_id=thread.id,
+            run_id=run.id
         )
-        for msg in messages:
-            if msg.role == 'assistant':
-                return msg.content[0]['text']['value']
+
+        if steps.data:
+            last_step = steps.data[-1]
+            if last_step.type == 'message_creation':
+                message_id = last_step.step_details.message_creation.message_id
+                message = client.beta.threads.messages.retrieve(
+                    thread_id=thread.id,
+                    message_id=message_id
+                )
+                return message.content[0]['text']['value']
 
         return "No response."
     except Exception as e:
-        return f"An error occurred: {str(e)}"
-    
+        return f"An error occurred: {str(e)}"  
+
 examples = [
     ["My Eucalyptus tree is struggling outside in the cold weather in europe"],
     ["My callatea house plant is yellowing."],