exec.py

import os
from flask_sqlalchemy import SQLAlchemy
from flask import Flask, render_template, request, url_for, flash
from werkzeug.utils import redirect

from ServerSide import predict_class

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db'
db = SQLAlchemy(app)

doctor_id = 0
patient_id = 0

class Patient(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    pname = db.Column(db.String(20), nullable=False)
    gender = db.Column(db.String(10), nullable=False)
    age = db.Column(db.Integer, nullable=False)
    pno = db.Column(db.Integer, nullable=False)
    result=db.Column(db.String(20),nullable=False)

    def __repr__(self):
        return f"Patient(id='{self.id}',name='{self.pname}',gender='{self.gender}',age='{self.age}',phone='{self.pno}',result='{self.result}')"

class Prescription(db.Model):
    prid = db.Column(db.Integer, primary_key=True)
    annot = db.Column(db.String(200), nullable=False)
    did = db.Column(db.Integer, db.ForeignKey('doctor.did'))
    doctor = db.relationship('Doctor', backref=db.backref('doctor', uselist=False))
    pid = db.Column(db.Integer, db.ForeignKey('patient.id'))
    patient = db.relationship('Patient', backref=db.backref('patient', uselist=False))

    def __repr__(self):
        return f"Prescription(id='{self.prid}',annotation='{self.annot}',p_id='{self.pid}',d_id='{self.did}')"

class Doctor(db.Model):
    dname = db.Column(db.String(20), nullable=False)
    did = db.Column(db.Integer, primary_key=True)
    dno = db.Column(db.Integer, nullable=False)

    def __repr__(self):
        return f"Doctor(name='{self.dname}', id='{self.did}', phone='{self.dno}')"

db.create_all()

@app.route('/')
def index():
    global doctor_id, patient_id
    doctor_id = 0
    patient_id = 0
    return render_template('index.html')

@app.route('/', methods = ['GET', 'POST'])
def predict_and_upload():
    if request.method == 'POST':
        file = request.files['images']
        path = os.getcwd()+'/static/tempdata/'+file.filename
        file.save(path)
        image_file = url_for('static', filename='tempdata/' + file.filename)
        output = predict_class(path)
        if output[0][0] == 1:
            labelans = "Normal"
        else:
            labelans = "Pneumonia"
        patient1 = Patient(pname=request.form['name'], gender=request.form['gender'], age=request.form['age'], pno=request.form['pno'], result=labelans)
        doctor1 = Doctor(dname=request.form['d_name'], did=request.form['d_id'], dno=request.form['d_mob'])
        db.session.add(patient1)
        db.session.commit()
        global patient_id, doctor_id
        patient_id = (Patient.query.all())[-1].id
        db.session.add(doctor1)
        db.session.commit()
        doctor_id = doctor1.did
        return render_template('index.html',predict=labelans,image_file=image_file,pname=patient1.pname,gender=patient1.gender,age=patient1.age,pno=patient1.pno,dname=doctor1.dname,dno=doctor1.dno,did=doctor1.did)

@app.route('/upload', methods = ['GET', 'POST'])
def upload_annot():
    if request.method=='POST':
        global doctor_id, patient_id
        prescription1 = Prescription(annot=request.form['annotation'],pid=patient_id,did=doctor_id)
        db.session.add(prescription1)
        db.session.commit()
        return redirect(url_for('index'))

if __name__ == '__main__':
    app.run(debug=True)