Update app.py

app.py

@@ -5,6 +5,7 @@ from news import News
 from datetime import datetime
 from os import listdir
 from web import Online_Scraper
+import requests
 
 app = Flask(__name__)
 
@@ -77,6 +78,26 @@ def Web():
     }
     return jsonify(result)
 
+@app.route('/imageneration', methods=['POST'])
+def IMGEN():
+    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}"}
+    
+    def query(payload):
+    	response = requests.post(API_URL, headers=headers, json=payload)
+    	return response.content
+    
+
+    result = {
+        'response': query({"inputs": prompt,})
+    }
+
+    return jsonify(result)
+    
 
 @app.route('/divyanshpizza', methods=['GET'])
 def get_counters():