Update app.py

app.py

@@ -15,7 +15,6 @@ app = Flask(__name__)
 
 @app.route('/mistral7b', methods=['POST'])
 def generate():
-    global counter
     # Get data from the request
     data = request.json
     prompt = data.get('prompt', '')
@@ -35,7 +34,6 @@ def generate():
 
 @app.route('/chatgpt', methods=['POST'])
 def chat():
-    global counter
     # Get data from the request
     data = request.json
     user_message = data.get('message', '')
@@ -54,7 +52,6 @@ def chat():
 
 @app.route('/news', methods=['GET'])
 def get_news():
-    global counter
     # Get data from the request
     key = request.args.get('key', '')
     cache_flag = request.args.get('cache', 'True').lower() == 'true'
@@ -78,10 +75,12 @@ def Web():
     }
     return jsonify(result)
 
-@app.route('/imageneration', methods=['GET'])
+@app.route('/imageneration', methods=['POST'])
 def IMGEN():
-    prompt = request.args.get('prompt', '')
-    key = request.args.get('key', '')
+    data = request.json
+    prompt = data.get('prompt', '')
+    key = data.get('key', '')
+
     
     API_URL = "https://api-inference.huggingface.co/models/OEvortex/HelpingAI-PixelCraft"
     headers = {"Authorization": f"Bearer {key}"}
@@ -95,7 +94,6 @@ def IMGEN():
 
 @app.route('/divyanshpizza', methods=['GET'])
 def get_counters():
-    global counter
     return jsonify(counter),jsonify({"data":str(listdir(r"static/data/"))})