app2.py

import json
import re
import time
import os
import ast
import logging
from flask import (
    Flask,
    render_template,
    request,
    jsonify,
    Response,
    stream_with_context,
)
import requests
from requests.adapters import HTTPAdapter, Retry

# Import process from process.py
from process import process
from pdb_helpers import mutate_residue, trim_pdb, get_dssp

app = Flask(__name__)


def stream_data(data):
    return f"data: {json.dumps(data)}\n\n"


@app.route("/", methods=["GET", "POST"])
def index():
    password = request.args.get("password")
    if password == "rebate-lemon-titanium":
        return render_template("index2.html")
    else:
        return render_template("login.html")


@app.route("/process", methods=["GET", "POST"])
def process_route():
    gene_name = request.args.get("gene_name")
    residue1 = request.args.get("residue1")
    position = int(request.args.get("position"))
    residue2 = request.args.get("residue2")

    # Validate input
    if not gene_name:
        return "Gene name is required", 400
    if not residue1:
        return "Residue 1 is required", 400
    if not position:
        return "Position is required", 400
    if not residue2:
        return "Residue 2 is required", 400

    def generate():
        yield stream_data({"message": "Processing request."})
        for data in process(gene_name, residue1, position, residue2):
            print(data)
            yield stream_data(data)
        yield stream_data({"type": "done", "message": "Done processing."})

    return Response(stream_with_context(generate()), content_type="text/event-stream")

@app.route("/mutate", methods=["POST"])
def mutate_route_route():
    data = request.get_json()  # Get JSON payload
    pdb_string = data.get("pdb_string")
    chain_id = data.get("chain_id", "A") or "A"
    position = int(data.get("position"))
    to_residue = data.get("to_residue")
    mutated = mutate_residue(pdb_string, chain_id, position, to_residue)
    return mutated

@app.route("/trim_pdb", methods=["POST"])
def trim_pdb_route():
    data = request.get_json()
    pdb_data = data.get("pdb_data")
    chains = data.get("chains")
    pdb_string = trim_pdb(pdb_data, chains)
    return pdb_string

@app.route("/dssp", methods=["POST"])
def dssp_route():
    data = request.get_json()
    pdb_string = data.get("pdb_string")
    dssp_data = get_dssp(pdb_string)
    return jsonify(dssp_data)