Merge pull request #35 from grant-baer/refactor-generate-image

Refactor image generation into backend call

Backend/backend.py

@@ -22,7 +22,17 @@
     "http://127.0.0.1:5001"  # This is where db_access.py is running.
 )
 
+@app.route("/generate_image", methods=["POST"])
+def generate_image():
+    data = request.get_json()
+    url = "https://imagegolf.io/api/generate"
+    url_data = {
+        "inputValue": data["prompt"]
+    }
+
+    r = requests.post(url, json=url_data)
 
+    return r.json()
 
 @app.route("/store_image", methods=["POST"])
 def store_image():
@@ -94,7 +104,7 @@ def login():
 
 
 @app.route("/create_user", methods=["POST"])
-def register():
+def create_user():
     print("received register request")
     print(request, request.data)
     return jsonify({"message": "No endpoint called create_user, perhaps you meant: /register"})
@@ -147,8 +157,6 @@ def register():
         return jsonify({"message": "Failed to create user!!"})
 
 
-
-
 if __name__ == "__main__":
     try:
         app.run(port=5000, debug=True)

Frontend/src/pages/create.js

@@ -1,4 +1,5 @@
 import React, { useState } from "react";
+import axios from "axios";
 import Image from "next/image";
 
 export default function Home() {
@@ -10,23 +11,16 @@ export default function Home() {
     e.preventDefault();
 
     try {
-      const response = await fetch("/api/generateURL", {
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-        },
-        body: JSON.stringify({
-          inputValue: text,
-        }),
-      });
+      const response = await axios.post(
+        "http://localhost:5000/generate_image",
+        {"prompt": text}
+      );
 
-      const data = await response.json();
-
-      if (data.url) {
-        setUrl(data.url);
+      if (response.data.output) {
+        setUrl(response.data.output);
         setImageAccepted(null); // Reset the decision state when a new image is fetched
-      } else {
-        console.error("Failed to get URL.");
+      } else if (response.data.error) {
+          console.error(response.data.error);
       }
     } catch (error) {
       console.error("Error:", error);