Update app.py

app.py

@@ -10,31 +10,46 @@ assistant_id=os.getenv('ASSISTANT_ID')
 client = openai.OpenAI(api_key=openai.api_key)
 
 def ask_openai(question):
-    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
-    )
-    
-    run = client.beta.threads.runs.retrieve(
-        thread_id=thread.id,
-        run_id=run.id,
-    )
-
-    messages = client.beta.threads.messages.list(
-        thread_id=thread.id,
-    )
-    for msg in messages:
-        if msg.role == 'assistant':
-            return msg.content[0]['text']['value']
-
-    return "No response."
+    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  
+        start_time = time.time()
+
+        while not response_received and time.time() - start_time < timeout:
+            run_status = client.beta.threads.runs.retrieve(
+                thread_id=thread.id,
+                run_id=run.id,
+            )
+            if run_status.status == 'completed':
+                response_received = True
+            else:
+                time.sleep(4) 
+
+        if not response_received:
+            return "Response timed out."
+
+        messages = client.beta.threads.messages.list(
+            thread_id=thread.id,
+        )
+        for msg in messages:
+            if msg.role == 'assistant':
+                return msg.content[0]['text']['value']
+
+        return "No response."
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
     
 examples = [
     ["My Eucalyptus tree is struggling outside in the cold weather in europe"],