app.py

from flask import *
from pyrebase import *
from dotenv import load_dotenv
import os
from authlib.integrations.flask_client import OAuth
import database as db
import carbon
import campaignAI
import os
import google.generativeai as genai
#-----------------setting up the app------------------

app = Flask(__name__)
app.config['SECRET_KEY'] = 'SECRET_KEY'
app.secret_key = 'SECRET_KEY' #generate a secret key and use it here in this virtual env
load_dotenv()

#------------------------Database setup-----------------------------
@app.before_request
def before_request():
    # Open database connection before each request
    db.connect_to_database()

@app.after_request
def after_request(response):
    # Close database connection after each request
    db.close_connection()
    return response

#--------------Authlib---------------
oauth = OAuth(app)
oauth.register(
    "club",
    client_id=os.getenv("OAUTH2_CLIENT_ID"),
    client_secret=os.getenv("OAUTH2_CLIENT_SECRET"),
    client_kwargs={"scope": "openid profile email",},
    server_metadata_url=f'{os.getenv("OAUTH2_META_URL")}',
)
#------------------------firebase settingc---------------------------

firebase_config = {
  'apiKey': os.getenv('APIKEY'),
  'authDomain': os.getenv('AUTHDOMAIN'),
  'projectId': os.getenv('PROJECTID'),
  'storageBucket': os.getenv('STORAGEBUCKET'),
  'messagingSenderId': os.getenv('MESSAGINGSENDERID'),
  'appId': os.getenv('APPID'),
  'measurementId': os.getenv('MEASUREMENTID'),
  'databaseURL': ""
}

firebase = initialize_app(firebase_config)
auth = firebase.auth() #auth for the user_token 

#------------------- Routing ------------------------

@app.route('/sign-up', methods=['GET','POST'])
def signup():
    if request.method == 'POST':

        email = request.form.get('newemail')
        password = request.form.get('newpassword')
        fname = request.form.get('firstname')
        lname = request.form.get('lastname')
        region = request.form.get('region')
        phone = request.form.get('phone')

        try:
            name = fname + " " + lname
            db.create_user(email,name,phone,0,region)
            user = auth.create_user_with_email_and_password(email, password)
            session['user'] = user
            return redirect('/')
        except Exception as e:
            print("Error During Signup:", e)
            flash("Something went wrong!",'error')
            return redirect('/sign-up')
            # Print error for debugging


    else:

        return render_template('login.html')


@app.route("/signin-google")
def googleCallback():
    
    try:
        # fetch access token and id token using authorization code
        token = oauth.clubsync.authorize_access_token()
        # print(token,"\n\n",type(token))

        token = dict(token)
        # print(json.dumps(token, indent = 4))
        # Extract necessary user data from the ID token
        personal = token.get('userinfo')
        user_info = {"name" : personal.get('name'),
                    "email" : personal.get('email'),
                    "idToken" : token.get('id_token')}
        # Set complete user information in the session
        session["user"] = user_info
        return redirect('/homepage') 
    except Exception as e:
        print("Error during google callback:", e)
        return redirect('/sign-up')


@app.route("/google-login")
def googleLogin():
    try:
        if "user" in session:
            abort(404)
        return oauth.clubsync.authorize_redirect(redirect_uri=url_for("googleCallback", _external=True))
    except Exception as e:
        flash("Error during google login:", "Failure")
        print(e)
        return redirect('/signup')

    

@app.route('/login', methods=['GET','POST'])
def login():

    if request.method == 'POST':
        try:
            email = request.form.get('email')
            password = request.form.get('password')

            user = auth.sign_in_with_email_and_password(email, password)
            name = db.fetch_user_name(email)
            user['name'] = name
            session['user'] = user
            return redirect('/')
        
        except Exception as e:
                    print("Error during login:", e)  # Print error for debugging
                    login_error = "Invalid email or password. Please try again."
                    flash(login_error)
                    return render_template('login.html')
        
    else:
        return render_template('login.html')
    

@app.route('/logout')
def logout():
    session.pop('user', None)
    session.clear() 
    flash("Logged out successfully!","Success")
    return redirect('/')




@app.route('/forgot-password', methods=['GET','POST'])
def forgotPassword():
    if request.method == 'POST':
        try:
            email = request.form.get('user_email')
            try:
                auth.send_password_reset_email(email)
                flash("Password reset link sent to your email!","Success")
                return redirect('/login')
            except Exception as e:
                print("Error during password reset:", e)
                flash("Something went wrong!","Error")
                return redirect('/login')
            
        except Exception as e:

            print("Error during password reset:", e)
            flash("Something went wrong!","Error")
            return redirect('/login')
        
    return render_template('forgotPassword.html')


@app.route('/')
def homepage():
    """
    
    """   
    if 'user' in session:

        #signed user
        
        user_id_token = session['user']
        
        try:
            
            email = session["user"].get('email')
            name = db.fetch_user_name(email)

            return render_template('index.html',name = name)
        

        except KeyError as e:
            flash(e,"Something went wrong!")
    else:
        return render_template('withoutindex.html')
    
@app.route('/ongoing-campaigns')
def campaigns():
    if "user" in session:
        return render_template('ongoingcmp.html')
    else:
        return redirect('/login')
    
@app.route('/campaign1')    
def campaign1():
    if "user" in session:
        try:
            campaignAI.waste_management_campaign()
            return redirect('/ongoing-campaigns')
        except Exception as e:
            print("Error during campaign creation:", e)
            flash("Something went wrong!", "Error")
            return redirect('/ongoing-campaigns')
    else:
        return redirect('/login')

@app.route('/campaign2')
def campaign2():
    if "user" in session:
        try:
            campaignAI.water_conservation_campaign()
            return redirect('/ongoing-campaigns')
        except Exception as e:
            print("Error during campaign creation:", e)
            flash("Something went wrong!", "Error")
            return redirect('/ongoing-campaigns')
    else:
        return redirect('/login')

@app.route('/campaign3')
def campaign3():
    if "user" in session:
        try:
            campaignAI.sustainable_development_campaign()
            return redirect('/ongoing-campaigns')
        except Exception as e:
            print("Error during campaign creation:", e)
            flash("Something went wrong!", "Error")
            return redirect('/ongoing-campaigns')
    else:
        return redirect('/login')

@app.route('/campaign4')    
def campaign4():
    if "user" in session:
        try:
            campaignAI.public_healthcare_campaign()
            return redirect('/ongoing-campaigns')
        except Exception as e:
            print("Error during campaign creation:", e)
            flash("Something went wrong!", "Error")
            return redirect('/ongoing-campaigns')
    else:
        return redirect('/login')
    
@app.route('/Green-O-Gram')
def space():
    if "user" in session:
        try:
            posts = db.fetch_social_posts()
            print(posts)
            return render_template('communitypost.html', posts=posts)
            
        except Exception as e:
            print("Error during fetching posts:", e)
            flash("Something went wrong!", "Error")
            return redirect('/')
    else:
        return redirect('/login')


@app.route('/post', methods=['GET','POST'])
def post():
    if "user" in session:
        if request.method == 'POST':
            try:
                post = request.form.get('caption')
                email = session["user"].get('email')
                title = request.form.get('title')
                db.create_social_post(email, post, "none", title)
                return redirect('/Green-O-Gram')
            except Exception as e:
                print("Error during post creation:", e)
                flash("Something went wrong!", "Error")
                return redirect('/Green-O-Gram')
        else:
            return render_template('photouplode.html')
    else:
        return redirect('/login')


    
@app.route('/carbon-footprint', methods=['GET','POST'])
def carbon_footprint():
    if "user" in session:
        user = session["user"]
        email = user["email"]
        if request.method == 'POST':
            try:
                electricity_kWh = float(request.form.get('electricity'))
                fuel_liters = float(request.form.get('fuel'))
                waste_kg = float(request.form.get('waste'))
                flight_hours = float(request.form.get('flight'))
                total_carbon_footprint = carbon.calculate_carbon_footprint(electricity_kWh, fuel_liters, waste_kg, flight_hours, 'india')
                print(total_carbon_footprint,type(total_carbon_footprint))
                db.update_user_cfp(email, total_carbon_footprint)
                try:
                    # Fetch table names and check if user's table exists
                    table_names = [table[0] for table in db.fetch_table_names()]
                    print(f"Fetched table names: {table_names}")

                    if f"{email}_cfp" in table_names:
                        db.track_user_cfp(email, total_carbon_footprint)
                    else:
                        db.create_user_cfp_table(email)
                        db.track_user_cfp(email, total_carbon_footprint)
                except Exception as e:
                    print(f"Error during carbon footprint tracking: {e}")
                    flash("Something went wrong!", "Error")
                    return redirect('/carbon-footprint')
                return render_template('footprint.html', total_carbon_footprint=total_carbon_footprint)
                return render_template('footprint.html', total_carbon_footprint=total_carbon_footprint)
            
            except Exception as e:
                print("Error during carbon footprint calculation:", e)
                flash("Something went wrong!","Error")
                return redirect('/carbon-footprint')
            return render_template('footprint.html')
        else:
            return render_template('footprint.html')

    else:
        return redirect('/login')
    
genai.configure(api_key=os.environ.get("API_KEY_GENAI"))


@app.route('/chat', methods=['POST'])
def chat():
    data = request.json
    user_message = data.get("message")
    print(user_message)

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

    try:
        # Define the Gemini AI model
        model = genai.GenerativeModel('gemini-1.5-flash')
        # Generate the response
        response = model.generate_content(user_message)
        bot_message = response.text.strip()
        print(bot_message)
    except Exception as e:
        return jsonify({"error": str(e)}), 500
   
    return jsonify({"message": bot_message})
    


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