Update app.py

app.py

@@ -1,43 +1,40 @@
 from flask import Flask, request, jsonify
-import json
-import base64
-import requests
-app = Flask(__name__)
-
-# GitHub credentials
-GITHUB_TOKEN = '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'  # Set your token in environment variables
-REPO_OWNER = 'hussein2000-oo'
-REPO_NAME = 'dbailloolloloolollhrthlnewrgnk'
-USER_FILE_NAME = 'user.json'
-
+from huggingface_hub import InferenceClient
 
-# Function to fetch user data from GitHub
-def fetch_user_data():
-    url = f'https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/contents/{USER_FILE_NAME}'
-    headers = {'Authorization': f'token {GITHUB_TOKEN}'}
-    response = requests.get(url, headers=headers)
+# Initialize the InferenceClient with your API key
+x="hf_uYlR"
+y="EMsDNbxQPJCSAHgwthrylHZZKKmGyg"
+u=x+y
+client = InferenceClient(api_key=f"{u}")
 
-    if response.status_code == 200:
-        content = response.json()
-        user_data = json.loads(base64.b64decode(content['content']).decode('utf-8'))
-        return user_data
-    else:
-        return None  # Return None if fetching fails
-
-
-@app.route('/get_security_questions', methods=['POST'])
-def get_security_questions():
-    data = request.json
-    username = data.get('username')
+# Create a Flask app
+app = Flask(__name__)
 
-    user_data = fetch_user_data()
+@app.route('/chat', methods=['POST'])
+def chat():
+    # Get user message from the request
+    user_message = request.json.get('message')
     
-    if user_data and username in user_data:
-        questions = user_data[username]['security_questions']
-        return jsonify({"security_questions": list(questions.keys())}), 200
-    else:
-        return jsonify({"error": "User  not found or data fetch failed."}), 404
-
+    # Check if the user message is provided
+    if not user_message:
+        return jsonify({"error": "No message provided"}), 400
+    
+    # Create a single message list for the request
+    messages = [{"role": "user", "content": user_message}]
+    
+    # Create the chat completion request with the current message
+    response = client.chat.completions.create(
+        model="Qwen/Qwen2.5-72B-Instruct", 
+        messages=messages, 
+        max_tokens=1024,
+        stream=False  # Set stream to False to get the full response at once
+    )
+    
+    # Get the assistant's response
+    assistant_message = response.choices[0].message.content
+    
+    # Return the assistant's response as JSON
+    return jsonify({"response": assistant_message})
 
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)