Update app.py

app.py

@@ -19,6 +19,7 @@ MESSAGE_API_URL = "https://aoamrnuwara.pythonanywhere.com/api/send-message"
 MESSAGE_API_KEY = "Seakp0683asppoit2"
 SCREENSHOT_API_URL = "https://corvo-ai-xx-sc.hf.space/capture"
 AI_API_URL = "https://corvo-ai-xx-gpt-5.hf.space/chat"
+AI_API_URL_O1 = "https://corvo-ai-xx-o1.hf.space/chat"
 ALERT_API_URL = "https://dooratre-alert.hf.space/monitor"
 
 # إعدادات المحاولات
@@ -48,7 +49,14 @@ def thinking_animation():
         print(".", end="", flush=True)
         time.sleep(0.5)
 
-def call_o1_ai_api(formatted_chat_history):
+def has_active_signal():
+    try:
+        signals_data = db_signals.fetch_json_from_github()
+        return bool(signals_data and signals_data.get("success") and signals_data.get("data"))
+    except Exception:
+        return False
+
+def call_o1_ai_api(formatted_chat_history, api_url=AI_API_URL):
     headers = {"Content-Type": "application/json"}
     payload = {"chat_history": formatted_chat_history}
     timeout = 600
@@ -57,7 +65,7 @@ def call_o1_ai_api(formatted_chat_history):
         try:
             print("AI THINKING", end="", flush=True)
             thinking_animation()
-            response = requests.post(AI_API_URL, headers=headers, data=json.dumps(payload), timeout=timeout)
+            response = requests.post(api_url, headers=headers, data=json.dumps(payload), timeout=timeout)
             response.raise_for_status()
             assistant_response = response.json().get("assistant_response", "No response received.")
             formatted_chat_history.append({"role": "assistant", "content": assistant_response})
@@ -988,7 +996,8 @@ def analysis_now():
         logger.info(f"🔔 Received alert message: {alert_message}")
 
         chat_history = build_chat_history(alert_message)
-        ai_response, updated_chat = call_o1_ai_api(chat_history)
+        api_url = AI_API_URL_O1 if has_active_signal() else AI_API_URL
+        ai_response, updated_chat = call_o1_ai_api(chat_history, api_url=api_url)
 
         tag_counts = count_xml_tags(ai_response)
         logger.info(f"📊 XML tags found: {tag_counts}")
@@ -1019,7 +1028,8 @@ def start_analysis():
     try:
         logger.info("🚀 Starting initial analysis...")
         chat_history = build_chat_history()
-        ai_response, updated_chat = call_o1_ai_api(chat_history)
+        api_url = AI_API_URL_O1 if has_active_signal() else AI_API_URL
+        ai_response, updated_chat = call_o1_ai_api(chat_history, api_url=api_url)
 
         tag_counts = count_xml_tags(ai_response)
         logger.info(f"📊 XML tags found: {tag_counts}")