Update app.py

app.py

@@ -7,13 +7,17 @@ from flask import Flask
 import scratchattach as sa
 
 # ==================== CONFIGURATION ====================
-# Use standard username/password authentication
+# Use the correct project ID (from your Scratch URL)
+PROJECT_ID = "1298553820"  # <-- Hardcoded, no need for environment variable
+
+# Use username/password authentication (more reliable than session string)
 SCRATCH_USER = os.environ.get("SCRATCH_USER")
 SCRATCH_PASS = os.environ.get("SCRATCH_PASS")
-PROJECT_ID = int(os.environ.get("PROJECT_ID", 0))
+
+# Your AI endpoint
 AI_URL = "https://16dvnk-spirit-kings-ai.hf.space/generate"
 
-# ==================== FULL PROXY LIST ====================
+# ==================== PROXY LIST ====================
 PROXY_LIST = [
     "http://8.219.97.248:80",
     "http://38.158.83.233:999",
@@ -151,10 +155,10 @@ class ProxyRotator:
 
 proxy_rotator = ProxyRotator(PROXY_LIST)
 
-# ==================== AI CALL WITH PROXY ROTATION ====================
+# ==================== AI CALL ====================
 def call_ai(prompt):
     print(f"[AI] Calling AI with prompt: {prompt[:50]}...")
-    for attempt in range(len(PROXY_LIST)):
+    for attempt in range(min(5, len(PROXY_LIST))):
         proxy = proxy_rotator.get_next_proxy()
         proxy_url = proxy.get("http", "")
         print(f"[AI] Attempt {attempt+1} using proxy {proxy_url}")
@@ -162,15 +166,15 @@ def call_ai(prompt):
             response = requests.post(AI_URL, json={"prompt": prompt}, proxies=proxy, timeout=30)
             if response.status_code == 200:
                 result = response.json().get("response", "Sorry, no response.")
-                print(f"[AI] Success, response length {len(result)}")
+                print(f"[AI] Success")
                 return result
             else:
-                print(f"[AI] Proxy {proxy_url} returned {response.status_code}")
+                print(f"[AI] Proxy failed: {response.status_code}")
                 proxy_rotator.mark_proxy_failed(proxy)
         except Exception as e:
-            print(f"[AI] Proxy {proxy_url} error: {e}")
+            print(f"[AI] Proxy error: {e}")
             proxy_rotator.mark_proxy_failed(proxy)
-    print("[AI] All proxies failed, trying direct...")
+    # Fallback direct
     try:
         response = requests.post(AI_URL, json={"prompt": prompt}, timeout=30)
         if response.status_code == 200:
@@ -181,75 +185,56 @@ def call_ai(prompt):
 
 # ==================== BRIDGE WORKER ====================
 def bridge_worker():
-    print("[BRIDGE] Starting bridge with standard authentication...")
+    print("[BRIDGE] Starting bridge with correct project ID...")
     print(f"[BRIDGE] Project ID: {PROJECT_ID}")
     
-    # Check if credentials are set
     if not SCRATCH_USER or not SCRATCH_PASS:
-        print("[BRIDGE] ERROR: SCRATCH_USER and SCRATCH_PASS environment variables must be set!")
-        return
-    
-    # Login once using standard method
-    print(f"[BRIDGE] Logging in as {SCRATCH_USER}...")
-    try:
-        session = sa.login(SCRATCH_USER, SCRATCH_PASS)
-        print("[BRIDGE] Login successful. Connecting to cloud variables...")
-        conn = session.connect_cloud(PROJECT_ID)
-        print(f"[BRIDGE] Connected to project {PROJECT_ID}")
-    except Exception as e:
-        print(f"[BRIDGE] Login/Connect failed: {e}")
-        import traceback
-        traceback.print_exc()
+        print("[BRIDGE] ERROR: SCRATCH_USER and SCRATCH_PASS environment variables not set!")
         return
     
-    last_prompt = "0"
     while True:
         try:
-            # Get current prompt value
-            current = conn.get_var("prompt")
-            if current is None:
-                current = "0"
-            print(f"[BRIDGE] Current prompt value: '{current}'")
+            print(f"[BRIDGE] Logging in as {SCRATCH_USER}...")
+            session = sa.login(SCRATCH_USER, SCRATCH_PASS)
+            print("[BRIDGE] Login successful. Connecting to cloud...")
+            cloud = session.connect_cloud(PROJECT_ID)
+            print(f"[BRIDGE] Connected to project {PROJECT_ID}")
             
-            if current != last_prompt and current != "0":
-                print(f"[BRIDGE] New prompt detected: {current}")
-                decoded = decode_text(current)
-                print(f"[BRIDGE] Decoded: {decoded}")
-                
-                ai_response = call_ai(decoded)
-                print(f"[BRIDGE] AI response: {ai_response[:100]}...")
-                
-                encoded_response = encode_text(ai_response)
-                conn.set_var("response", encoded_response)
-                print("[BRIDGE] Response sent to Scratch")
-                
-                last_prompt = current
-            else:
-                print("[BRIDGE] No change, sleeping...")
+            # Set up event listener for real-time updates
+            events = cloud.events()
+            print("[BRIDGE] Event listener active. Waiting for 'prompt' changes...")
+            
+            @events.event
+            def on_set(activity):
+                if activity.var == "prompt" and activity.value != "0":
+                    print(f"[EVENT] Received prompt: {activity.value}")
+                    decoded = decode_text(activity.value)
+                    print(f"[EVENT] Decoded: {decoded}")
+                    ai_response = call_ai(decoded)
+                    print(f"[EVENT] AI response: {ai_response[:100]}...")
+                    encoded_response = encode_text(ai_response)
+                    cloud.set_var("response", encoded_response)
+                    print("[EVENT] Response sent to Scratch")
             
-            time.sleep(0.3)  # Poll every 0.3 seconds
+            events.start()
             
+            # Keep the thread alive
+            while True:
+                time.sleep(1)
+                
         except Exception as e:
-            print(f"[BRIDGE] Error in loop: {e}")
+            print(f"[BRIDGE] Error: {e}")
             import traceback
             traceback.print_exc()
             print("[BRIDGE] Reconnecting in 10 seconds...")
             time.sleep(10)
-            # Attempt to reconnect
-            try:
-                session = sa.login(SCRATCH_USER, SCRATCH_PASS)
-                conn = session.connect_cloud(PROJECT_ID)
-                print("[BRIDGE] Reconnected successfully")
-                last_prompt = "0"  # Reset to avoid duplicate processing
-            except Exception as reconnect_error:
-                print(f"[BRIDGE] Reconnect failed: {reconnect_error}")
 
 # ==================== FLASK APP ====================
 app = Flask(__name__)
 
 @app.route('/')
 def health():
-    return "Scratch AI Bridge is running with standard authentication."
+    return "Scratch AI Bridge is running with correct project ID."
 
 if __name__ == '__main__':
     print("[MAIN] Starting bridge thread...")