Created app.py

app.py

@@ -0,0 +1,172 @@
+import os
+import json
+import uuid
+import requests
+import time
+from flask import Flask, request, Response, jsonify
+
+app = Flask(__name__)
+
+# ==============================================================================
+# CONFIG
+# ==============================================================================
+# Hardcoded credentials from hi.py
+# You can also use os.environ.get("COOKIE_VALUE") if you add secrets in Space settings
+RAW_COOKIE = "cf_clearance=OKEc8I4zkAdc_cYAx8zSVCOSuOSV2_kVsqBE7aaIIeM-1770044471-1.2.1.1-gaa4G052d_XBWdr.OJMsLGakekbPev1RiqMPA0vukct0SIspf0SInwXuY3yJeadggphdWtqYgRodnKKg9WfzOJh2EUtNY53fnnkHhs5ntywLMuk73FZTEI5NMim_FxjJ2xQXXRFxOAuacHCESyTeC96oOr_FrTdu0gZ0hJX3z5Wurw2h2KGEMqsAq737XtG2G6qUYckk10v4uWxLyFmEZHIiAajtwdh6F.ImIFJN_.Y; __Secure-better-auth.state=M-xPebcuX_QNYD573nAaocuMydAVHN5i.vRMOdmB9Lyb6Pyw2D7NejhtM%2FuWyWoNLGRnJv3Zou8k%3D; __Secure-better-auth.session_token=JQfRDZNrxwqaQ5kImpLqwm16hBFLPaQV.IUrxupoeprZ1fq7HftehR%2FybojAB%2FzcY%2BjT7KAiKKw8%3D; __Secure-better-auth.session_data=eyJzZXNzaW9uIjp7InNlc3Npb24iOnsiZXhwaXJlc0F0IjoiMjAyNi0wMi0wOVQxNTowMTozMS45MDBaIiwidG9rZW4iOiJKUWZSRFpOcnh3cWFRNWtJbXBMcXdtMTZoQkZMUGFRViIsImNyZWF0ZWRBdCI6IjIwMjYtMDItMDJUMTU6MDE6MzEuOTAwWiIsInVwZGF0ZWRBdCI6IjIwMjYtMDItMDJUMTU6MDE6MzEuOTAwWiIsImlwQWRkcmVzcyI6IjEwNC4yMy4xOTAuMTQ4IiwidXNlckFnZW50IjoiTW96aWxsYS81LjAgKEFuZHJvaWQgMTQ7IE1vYmlsZTsgcnY6MTQ3LjApIEdlY2tvLzE0Ny4wIEZpcmVmb3gvMTQ3LjAiLCJ1c2VySWQiOiIyZTRjYmVlYS03ODQ2LTQwYTktYmI0Yy1kNzQ3MWEzMWJiMTgiLCJpbXBlcnNvbmF0ZWRCeSI6bnVsbCwiaWQiOiJkMGU4OTZmZi04ZDhmLTQ4Y2MtOTIwZS05ZjBjYjc4OWM4NDUifSwidXNlciI6eyJuYW1lIjoiQnJ1ayBHZXRhY2hldyIsImVtYWlsIjoiYnJ1a2c5NDE5QGdtYWlsLmNvbSIsImVtYWlsVmVyaWZpZWQiOnRydWUsImltYWdlIjoiaHR0cHM6Ly9saDMuZ29vZ2xldXNlcmNvbnRlbnQuY29tL2EvQUNnOG9jSjBHeWpPWFVmSXVpVTIzeTdGcE53WExKcHdQNHRQTV95R3RhVkp2aU9HTjBITW1RPXM5Ni1jIiwiY3JlYXRlZEF0IjoiMjAyNi0wMi0wMlQxNTowMTozMS44NDZaIiwidXBkYXRlZEF0IjoiMjAyNi0wMi0wMlQxNTowMTozMS44NDZaIiwicm9sZSI6ImVkaXRvciIsImJhbm5lZCI6ZmFsc2UsImJhblJlYXNvbiI6bnVsbCwiYmFuRXhwaXJlcyI6bnVsbCwiaWQiOiIyZTRjYmVlYS03ODQ2LTQwYTktYmI0Yy1kNzQ3MWEzMWJiMTgifX0sImV4cGlyZXNBdCI6MTc3MDA1Mzg4MjkzNiwic2lnbmF0dXJlIjoiZFFTQkVZbmFlWTV2MGM1ZE8zbURWU1RSRlVDWURrMlFYN1J0bFItdVpObyJ9"
+
+HEADERS = {
+    "Accept": "*/*",
+    "Accept-Encoding": "gzip, deflate, br, zstd",
+    "Accept-Language": "en-ER,am-ER;q=0.9",
+    "Connection": "keep-alive",
+    "Content-Type": "application/json",
+    "Origin": "https://www.cognixai.co",
+    "Referer": "https://www.cognixai.co/chat",
+    "Sec-Fetch-Dest": "empty",
+    "Sec-Fetch-Mode": "cors",
+    "Sec-Fetch-Site": "same-origin",
+    "User-Agent": "Mozilla/5.0 (Android 14; Mobile; rv:147.0) Gecko/147.0 Firefox/147.0"
+}
+
+def parse_cookies(cookie_string):
+    """Parses cookie string into dict"""
+    cookies = {}
+    items = cookie_string.split(';')
+    for item in items:
+        if '=' in item:
+            parts = item.strip().split('=', 1)
+            if len(parts) == 2:
+                cookies[parts[0]] = parts[1]
+    return cookies
+
+COOKIES = parse_cookies(RAW_COOKIE)
+
+# ==============================================================================
+# LOGIC
+# ==============================================================================
+
+def generate_openai_chunk(content, model, finish_reason=None):
+    """Create OpenAI-compatible stream chunk"""
+    chunk_id = f"chatcmpl-{uuid.uuid4().hex[:24]}"
+    timestamp = int(time.time())
+    
+    return {
+        "id": chunk_id,
+        "object": "chat.completion.chunk",
+        "created": timestamp,
+        "model": model,
+        "choices": [{
+            "index": 0,
+            "delta": {"content": content} if content else {},
+            "finish_reason": finish_reason
+        }]
+    }
+
+@app.route('/v1/chat/completions', methods=['POST'])
+def chat_completions():
+    try:
+        data = request.json
+        messages = data.get('messages', [])
+        model = data.get('model', 'Gemini 3 Pro Thikning') # Default from hi.py
+        stream = data.get('stream', False)
+
+        # 1. Extract User Prompt
+        last_user_content = ""
+        for msg in reversed(messages):
+            if msg['role'] == 'user':
+                content = msg['content']
+                if isinstance(content, list):
+                    last_user_content = " ".join([p.get('text', '') for p in content if p.get('type') == 'text'])
+                else:
+                    last_user_content = str(content)
+                break
+        
+        if not last_user_content:
+            return jsonify({"error": "No user message found"}), 400
+
+        # 2. Build Payload for Cognix
+        payload = {
+            "id": str(uuid.uuid4()),
+            "chatModel": {
+                "provider": "google",
+                "model": model 
+            },
+            "toolChoice": "auto",
+            "allowedAppDefaultToolkit": ["code", "visualization"],
+            "message": {
+                "role": "user",
+                "parts": [{"type": "text", "text": last_user_content}],
+                "id": str(uuid.uuid4())
+            },
+            "imageTool": {},
+            "attachments": []
+        }
+
+        # 3. Stream Response
+        def generate():
+            url = "https://www.cognixai.co/api/chat"
+            print(f"Sending to Cognix: {last_user_content[:50]}...")
+            
+            with requests.post(url, headers=HEADERS, cookies=COOKIES, json=payload, stream=True) as response:
+                if response.status_code != 200:
+                    yield f"data: {json.dumps({'error': f'Upstream Error: {response.status_code}'})}\n\n"
+                    return
+
+                for line in response.iter_lines():
+                    if line:
+                        decoded = line.decode('utf-8')
+                        if decoded.startswith("data: "):
+                            try:
+                                json_data = json.loads(decoded[6:])
+                                delta_content = None
+                                
+                                # Logic from hi.py
+                                if json_data.get("type") == "reasoning-delta":
+                                    delta_content = json_data.get("delta", "")
+                                elif json_data.get("type") == "text-delta":
+                                    delta_content = json_data.get("delta", "")
+                                elif json_data.get("type") == "error":
+                                    print(f"Stream Error: {json_data}")
+                                
+                                if delta_content:
+                                    chunk = generate_openai_chunk(delta_content, model)
+                                    yield f"data: {json.dumps(chunk)}\n\n"
+                                    
+                            except Exception as e:
+                                print(f"Parse error: {e}")
+                                continue
+                
+                # End of stream
+                final_chunk = generate_openai_chunk(None, model, "stop")
+                yield f"data: {json.dumps(final_chunk)}\n\n"
+                yield "data: [DONE]\n\n"
+
+        if stream:
+            return Response(generate(), mimetype='text/event-stream')
+        else:
+            # Non-streaming fallback (collect all)
+            full_response = ""
+            for chunk_str in generate():
+                if chunk_str.startswith("data: [DONE]"): break
+                if chunk_str.startswith("data: "):
+                    try:
+                        chunk = json.loads(chunk_str[6:])
+                        if 'choices' in chunk:
+                            content = chunk['choices'][0]['delta'].get('content', '')
+                            full_response += content
+                    except: pass
+            
+            return jsonify({
+                "id": str(uuid.uuid4()),
+                "object": "chat.completion",
+                "created": int(time.time()),
+                "model": model,
+                "choices": [{"index": 0, "message": {"role": "assistant", "content": full_response}, "finish_reason": "stop"}]
+            })
+
+    except Exception as e:
+        print(f"Error: {e}")
+        return jsonify({"error": str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)