web.py

import os
import PIL
import json
from requests import get
from google import genai
from dotenv import load_dotenv
from Functions.Data import MK_File
from Functions.Files import read_file, upload_video
from werkzeug.utils import secure_filename
from Functions.Main_Response import process_prompt
from flask import Flask, render_template, request, jsonify, url_for, send_from_directory


load_dotenv()
client = genai.Client(api_key=os.getenv("GENAI_API_KEY"))
app = Flask(__name__)
log_file = MK_File()
# Raw URL of the README file
url = "https://raw.githubusercontent.com/siMobin/Gemini-Echo/a69734c328b3ae6c0df35b704fe62671fcde65e7/README.md"
pre = get(url)

# Set the upload folder for media files
UPLOAD_FOLDER = "static/uploads"
app.config["UPLOAD_FOLDER"] = UPLOAD_FOLDER

with open("instructions/keywords.json") as f:
    keywords = json.load(f)

with open("instructions/commands.json") as f:
    commands = json.load(f)


# Check if the request was successful
if pre.status_code == 200:
    pre_1 = "\n".join(pre.text.split("\n")[8:19])
else:
    pass

greeting = commands["web_startup"]
response = process_prompt(greeting, log_file)


@app.route("/")
def index():
    return render_template("index.html", greeting=response, pre_1=pre_1)


@app.route("/get-keywords")
def get_keywords():
    return send_from_directory("instructions", "keywords.json")


@app.route("/chat", methods=["POST"])
def chat():
    user_message = request.form.get("message")
    media_file = request.files.get("media")

    # Debugging log to check if message and media file are received
    print(f"Received user message: {user_message}")
    print(f"Received media file: {media_file}")

    if not user_message:
        return jsonify({"error": "No message provided"}), 400

    # Initialize media variables
    image = None
    media_audio = None
    media_video = None
    media_path = None

    # Check if the file is valid and allowed
    if media_file:
        filename = secure_filename(media_file.filename)
        file_extension = f".{filename.rsplit('.', 1)[1].lower()}"

        # Debug
        print(f"File extension: {file_extension}")  # Debugging line
        # Save the file to the server
        file_path = os.path.join(UPLOAD_FOLDER, filename)
        media_file.save(file_path)
        media_path = url_for("static", filename=f"uploads/{filename}")  # Adjusted path

        # Check file type and set the appropriate file path variable
        if file_extension in keywords["images"]:
            image = PIL.Image.open(file_path)
        elif file_extension in keywords["audios"]:
            media_audio = client.files.upload(file=file_path)
        elif file_extension in keywords["videos"]:
            media_video = upload_video(file_path, client)
        elif file_extension in keywords["documents"]:
            print(file_path)
            info = read_file(file_path)
            print(f"Read file content: {info}")  # Check what info contains
            user_message = info + "\n\n\n" + user_message
        else:
            print(f"Unsupported file type: {file_extension}")  # Debugging line
            return jsonify({"error": "Unsupported file type"}), 400
    else:
        print("No media file uploaded or unsupported file type.")  # Debugging line

    response = process_prompt(
        user_message,
        log_file,
        image=image,
        media_audio=media_audio,
        media_video=media_video,
    )

    return jsonify({"response": response, "media": media_path})


if __name__ == "__main__":
    app.run(host=os.getenv("host"), port=os.getenv("port"), debug=os.getenv("debug"))