Update app.py

app.py

@@ -1,82 +1,100 @@
-import requests
 from flask import Flask, request, jsonify
-from datetime import datetime
+import requests
+from bs4 import BeautifulSoup
+import json
 
 app = Flask(__name__)
 
-# Your PythonAnywhere API endpoint
-PYTHONANYWHERE_URL = "https://omarnuwara.pythonanywhere.com/get-response"
+# Step 1: Fetch the authenticity_token and commitOid from the GitHub edit page
+def fetch_authenticity_token_and_commit_oid():
+    url = "https://github.com/omarnuwrar/api/edit/main/user.json"
 
-# URL of GitHub-hosted user database
-USER_DB_URL = "https://raw.githubusercontent.com/omarnuwrar/api/refs/heads/main/user.json"
+    headers = {
+        "cookie": "_octo=GH1.1.1509769180.1721774314; _device_id=0038e28d4f7d4f9baf8f76b6b9fb8980; _gh_sess=XxMNHmqq58kaNJpj3HKllufdyrUNNjxg3%2FW7rJNFwdwO%2FHbrcT2QcSEdzfhBTWdu3PrkDAVy6vs46R6I5kEENdJ%2Bhixu1gb5HiLv1uKh4lKCiwbG16TlJfbkGXUhvou9zTc9Bszn%2F4HOPD0%2FlG%2BauN1hu2MYcajmfMV%2Fb2R%2FvM8tABljOethGPx5CzrGE1b8EgZxMeOP5wNBrZH4KHPIAD6IAb2z7j%2FJaXHAImK9tmU5CIb5Q9fi9FH6l%2BjWdmSYXekDVHzrKK8mXmO2%2B8N%2BaG0r%2BqoBACKdpj3rbC9x1%2BSKMZjSmu0uNrCmJAPymukisTHFRVtuMikNtEZtVx8QYH%2F9j4hDPs6fOE7wAegPTG7xYeBM1QCNJgiHHGHzoawBFbs0DtEcZwIvRQgVCXYpE0YIUbp6LCEoVxxQeSiTYrBZcoXZ%2F%2BFvw7J7mZSG5pBfo%2FuZ6xQSblzlWpzg--SXX57IvnIMkLdZCr--s5FjBmuKtrEf71mhHx1g2g%3D%3D; saved_user_sessions=155741452%3Avhq2fotRhM6ixLZuU8plCTagdBxOI9UGuu5XwPf569UWHPOg; user_session=vhq2fotRhM6ixLZuU8plCTagdBxOI9UGuu5XwPf569UWHPOg; __Host-user_session_same_site=vhq2fotRhM6ixLZuU8plCTagdBxOI9UGuu5XwPf569UWHPOg; logged_in=yes; dotcom_user=omarnuwrar; color_mode=%7B%22color_mode%22%3A%22auto%22%2C%22light_theme%22%3A%7B%22name%22%3A%22light%22%2C%22color_mode%22%3A%22light%22%7D%2C%22dark_theme%22%3A%7B%22name%22%3A%22dark%22%2C%22color_mode%22%3A%22dark%22%7D%7D; cpu_bucket=lg; preferred_color_mode=light; tz=Africa%2FTripoli",
+        "if-none-match": 'W/"2ff86bd1792cfee5ed79ee070b3b46de"',
+        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+        "x-github-target": "dotcom",
+        "x-react-router": "json",
+        "x-requested-with": "XMLHttpRequest",
+    }
+
+    response = requests.get(url, headers=headers)
 
-def get_users_from_db():
-    """Fetch user data from the GitHub JSON file."""
-    response = requests.get(USER_DB_URL)
     if response.status_code == 200:
-        return response.json()
-    else:
-        raise Exception("Failed to fetch the user database.")
-
-def validate_token(token):
-    """Validate the provided token."""
-    users = get_users_from_db()
-    for user in users:
-        if user["token"] == token:
-            # Parse the expiration time from the database
-            expiration_time = datetime.fromisoformat(user["token_expiration_time"])
-            
-            # Get the current time in the same format
-            current_time = datetime.now()
-
-            print(f"Token Expiration Time: {expiration_time}")
-            print(f"Current Time: {current_time}")
-
-            # Check if the token has expired
-            if current_time < expiration_time:
-                return True, None  # Token is valid
-            else:
-                return False, "Token has expired."
-    return False, "Token not found."
-
-@app.route('/chat', methods=['POST'])
-def chat():
-    try:
-        # Ensure the request contains JSON with "message" and "token" keys
-        if not request.is_json:
-            return jsonify({"error": "Request must be in JSON format."}), 400
-
-        user_input = request.json.get("message", "")
-        token = request.json.get("token", "")
-
-        if not user_input:
-            return jsonify({"error": "No message provided in the request."}), 400
-
-        if not token:
-            return jsonify({"error": "No token provided in the request."}), 400
-
-        # Validate token
-        token_valid, token_message = validate_token(token)
-        if not token_valid:
-            return jsonify({"error": "Invalid token.", "details": token_message}), 401
-
-        # Forward the message to the PythonAnywhere endpoint
-        data = {"message": [{"role": "user", "content": user_input}]}
-        response = requests.post(PYTHONANYWHERE_URL, json=data)
-
-        if response.status_code == 200:
-            response_json = response.json()
-            ai_response = response_json.get("response", "No 'response' key found in the JSON.")
-            ai_response = ai_response.encode('latin1').decode('utf-8', 'ignore')
-            return jsonify({"response": ai_response})
+        soup = BeautifulSoup(response.text, 'html.parser')
+        script_tag = soup.find("script", {"type": "application/json", "data-target": "react-app.embeddedData"})
+
+        if script_tag:
+            try:
+                json_data = json.loads(script_tag.string.strip())
+                authenticity_token = json_data["payload"]["csrf_tokens"]["/omarnuwrar/api/tree-save/main/user.json"]["post"]
+                commit_oid = json_data["payload"]["webCommitInfo"]["commitOid"]
+                return authenticity_token, commit_oid
+            except (KeyError, json.JSONDecodeError) as e:
+                print(f"Error: Failed to extract data. Details: {str(e)}")
+                return None, None
         else:
-            return jsonify({
-                "error": "Error from PythonAnywhere",
-                "details": response.text
-            }), response.status_code
+            print("Error: Could not find the required <script> tag.")
+            return None, None
+    else:
+        print(f"Error: Failed to fetch the page. Status code: {response.status_code}")
+        return None, None
+
+# Step 2: Send the POST request to update the user.json file
+def update_user_json_file(authenticity_token, commit_oid, new_content):
+    url = "https://github.com/omarnuwrar/api/tree-save/main/user.json"
+
+    headers = {
+        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+        "x-requested-with": "XMLHttpRequest",
+        "github-verified-fetch": "true",
+        "content-type": "application/x-www-form-urlencoded",
+        "cookie": "_octo=GH1.1.1509769180.1721774314; _device_id=0038e28d4f7d4f9baf8f76b6b9fb8980; _gh_sess=XxMNHmqq58kaNJpj3HKllufdyrUNNjxg3%2FW7rJNFwdwO%2FHbrcT2QcSEdzfhBTWdu3PrkDAVy6vs46R6I5kEENdJ%2Bhixu1gb5HiLv1uKh4lKCiwbG16TlJfbkGXUhvou9zTc9Bszn%2F4HOPD0%2FlG%2BauN1hu2MYcajmfMV%2Fb2R%2FvM8tABljOethGPx5CzrGE1b8EgZxMeOP5wNBrZH4KHPIAD6IAb2z7j%2FJaXHAImK9tmU5CIb5Q9fi9FH6l%2BjWdmSYXekDVHzrKK8mXmO2%2B8N%2BaG0r%2BqoBACKdpj3rbC9x1%2BSKMZjSmu0uNrCmJAPymukisTHFRVtuMikNtEZtVx8QYH%2F9j4hDPs6fOE7wAegPTG7xYeBM1QCNJgiHHGHzoawBFbs0DtEcZwIvRQgVCXYpE0YIUbp6LCEoVxxQeSiTYrBZcoXZ%2F%2BFvw7J7mZSG5pBfo%2FuZ6xQSblzlWpzg--SXX57IvnIMkLdZCr--s5FjBmuKtrEf71mhHx1g2g%3D%3D; saved_user_sessions=155741452%3Avhq2fotRhM6ixLZuU8plCTagdBxOI9UGuu5XwPf569UWHPOg; user_session=vhq2fotRhM6ixLZuU8plCTagdBxOI9UGuu5XwPf569UWHPOg; __Host-user_session_same_site=vhq2fotRhM6ixLZuU8plCTagdBxOI9UGuu5XwPf569UWHPOg; logged_in=yes; dotcom_user=omarnuwrar; color_mode=%7B%22color_mode%22%3A%22auto%22%2C%22light_theme%22%3A%7B%22name%22%3A%22light%22%2C%22color_mode%22%3A%22light%22%7D%2C%22dark_theme%22%3A%7B%22name%22%3A%22dark%22%2C%22color_mode%22%3A%22dark%22%7D%7D; cpu_bucket=lg; preferred_color_mode=light; tz=Africa%2FTripoli",
+    }
+
+    payload = {
+        "message": "Update user.json",
+        "placeholder_message": "Update user.json",
+        "description": "",
+        "commit-choice": "direct",
+        "target_branch": "main",
+        "quick_pull": "",
+        "guidance_task": "",
+        "commit": commit_oid,
+        "same_repo": "1",
+        "pr": "",
+        "content_changed": "true",
+        "filename": "user.json",
+        "new_filename": "user.json",
+        "value": new_content,
+        "authenticity_token": authenticity_token,
+    }
+
+    response = requests.post(url, headers=headers, data=payload)
 
-    except Exception as e:
-        return jsonify({"error": "An error occurred", "details": str(e)}), 500
+    if response.status_code == 200:
+        return {"success": True, "message": "user.json has been updated!"}
+    else:
+        return {"success": False, "message": f"Request failed with status code {response.status_code}", "details": response.text}
+
+
+# API Endpoint to update user.json
+@app.route('/update_user_json', methods=['POST'])
+def update_user_json():
+    data = request.json
+
+    if not data or 'content' not in data:
+        return jsonify({"success": False, "message": "Invalid request. 'content' field is required."}), 400
+
+    # Step 1: Fetch the authenticity_token and commitOid
+    authenticity_token, commit_oid = fetch_authenticity_token_and_commit_oid()
+
+    if authenticity_token and commit_oid:
+        # Step 2: Perform the file update
+        result = update_user_json_file(authenticity_token, commit_oid, data['content'])
+        return jsonify(result)
+    else:
+        return jsonify({"success": False, "message": "Failed to fetch required tokens."}), 500
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)