Add application file

Dockerfile

@@ -0,0 +1,20 @@
+# Use the official Python image
+FROM python:3.9-slim
+
+# Set the working directory
+WORKDIR /app
+
+# Copy the requirements file into the image
+COPY requirements.txt requirements.txt
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of your application code into the image
+COPY . .
+
+# Expose the port your application will run on
+EXPOSE 7860
+
+# Command to run your application
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,41 @@
+from flask import Flask, jsonify, request
+from flask_cors import CORS
+
+app = Flask(__name__)
+ 
+# Enable CORS for specific origins
+CORS(app, resources={r"api/predict/*": {"origins": ["http://localhost:3000", "https://main.dbn2ikif9ou3g.amplifyapp.com"]}})
+
+@app.route("/", methods=["GET"])
+def handle_get_request():
+    # Get the 'message' parameter from the query string
+    message = request.args.get("message", "No message provided.")
+    
+    # Return a JSON response including the received message
+    return jsonify({"message": message, "status": "GET request successful!"})
+
+@app.route("/send_message", methods=["POST"])
+def handle_post_request():
+    # Get the JSON data from the request
+    data = request.get_json()
+
+    # Check if data is None
+    if data is None:
+        return jsonify({"error": "No JSON data provided"}), 400
+
+    # Extract the 'inputs' and 'authtoken' from the JSON data
+    message = data.get("inputs", "No message provided.")
+    authtoken = data.get("authtoken", "No auth token provided.")
+    
+    # Return a JSON response including the received message and auth token
+    return jsonify({
+        "received_message": message,
+        "received_authtoken": authtoken,
+        "status": "POST request successful!"
+    })
+
+# Note: Remove the app.run() call to let Hugging Face handle it
+# Launch the interface
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)

requirements.txt

@@ -0,0 +1,5 @@
+flask
+flask_cors
+transformers
+torch
+huggingface-hub