Update flask_app.py

flask_app.py

@@ -1,10 +1,11 @@
    
 import os
-from flask import Flask, flash, request, redirect, url_for, session
+from flask import Flask, flash, request, redirect, url_for, session, jsonify
 from flask_session import Session
 from werkzeug.utils import secure_filename
 from example_inference import example_inference
 from flask import send_from_directory
+from gradio_client import Client
 import base64
 
 UPLOAD_FOLDER = 'images'
@@ -16,7 +17,49 @@ app.config["SESSION_TYPE"] = "filesystem"
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
 Session(app)
- 
+
+client = Client("stabilityai/stable-diffusion-3-medium")
+
+
+def generate_logo(prompt):
+  """Generates a logo using the provided prompt."""
+
+  result = client.predict(
+      prompt=prompt,
+      negative_prompt="",
+      seed=0,
+      randomize_seed=True,
+      width=1024,
+      height=1024,
+      guidance_scale=7, 
+      num_inference_steps=50,
+      api_name="/infer",
+  )
+  return result
+
+
+@app.route('/logo', methods=['POST'])
+def generate_response():
+    try:
+        data = request.get_json()
+        
+        if 'prompt' in data :
+            prompt = data['prompt']
+            
+            response  = generate_logo(prompt)
+            print(response, 'response')
+            fileUrl = f'https://stabilityai-stable-diffusion-3-medium.hf.space/file={response[0]}'
+
+            return jsonify(fileUrl)
+
+        else:
+            return jsonify({"error": "Missing required parameters"}), 400
+
+    except Exception as e:
+        return jsonify({"Error": str(e)}), 500 
+
+
+
 def allowed_file(filename):
     return '.' in filename and \
            filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS