Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, send_from_directory, url_for, flash,redirect
+from flask import Flask, render_template, request, send_from_directory, url_for, flash,redirect,session
 import requests
 import base64
 import json
@@ -315,6 +315,26 @@ def create_account():
 
     return render_template('sign-up.html')
 
+@app.route("/")
+def home():
+    global conversation_history
+    conversation_history = []
+    
+    # Retrieve user information from the session
+    user_info = session.get('user_info', {})
+    account_name = user_info.get('account_name')
+    first = user_info.get('first')
+    last = user_info.get('last')
+    token2 = user_info.get('token2')
+    greet = user_info.get('greet')
+    
+    if not account_name:
+         conversation_history = []
+         return redirect(url_for('login'))
+    
+    return render_template('ai-chat-bot.html', account_name=account_name, greet=greet, first=first, last=last, token2=token2)
+
+
 
 
 
@@ -326,43 +346,54 @@ def login():
         password = request.form.get("password")
         message = process_account_action("login", account_name, password)
         flash(message)
-        if "Login successful." in message :
-          ############### GET INFO ABOUT USER ###############################################
-          url = f"https://api.github.com/repos/omarnuwrar/{repository_name}/contents/user.json"
-          headers = {
-                         "Authorization": f"token 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",
-                         "Accept": "application/vnd.github.v3+json"
-                                                     }
-          response = requests.get(url, headers=headers)
-          if response.status_code == 200:
-            file_info = response.json()
-            file_content = base64.b64decode(file_info['content']).decode()
-            users = eval(file_content)  # Convert the string to a list of dictionaries
-
-            target_username = account_name
-
-            target_user = None
-            for user in users:
-              if user['username'] == target_username:
-               target_user = user
-               break
+        if "Login successful." in message:
+            ############### GET INFO ABOUT USER ###############################################
+            url = f"https://api.github.com/repos/omarnuwrar/{repository_name}/contents/user.json"
+            headers = {
+                "Authorization": f"token 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",
+                "Accept": "application/vnd.github.v3+json"
+            }
+            response = requests.get(url, headers=headers)
+            if response.status_code == 200:
+                file_info = response.json()
+                file_content = base64.b64decode(file_info['content']).decode()
+                users = eval(file_content)  # Convert the string to a list of dictionaries
+
+                target_username = account_name
+
+                target_user = None
+                for user in users:
+                    if user['username'] == target_username:
+                        target_user = user
+                        break
+
+                first = target_user['first_name']
+                last = target_user['last_name']
+                token2 = target_user['token']
+                user_name.append(first)
+
+            ############## END INFO ABOUT USER #################################################
+
+            # Initialize the user input and get the assistant's response
+            user_input = f"user name : {first} \n \n message : أهلا"
+            greet = get_assistant_response(user_input)
+
+            # Store user details in the session
+            session['user_info'] = {
+                'account_name': account_name,
+                'first': first,
+                'last': last,
+                'token2': token2,
+                'greet': greet
+            }
+
+            # Redirect to the main route
+            return redirect(url_for('home'))
 
-    
-            first =  target_user['first_name']
-            last =  target_user['last_name']
-            token2 =  target_user['token']
-            user_name.append(first)
-
-          ##############END INFO ABOUT USER #################################################
-          conversation_history = []
-          image_chat = []
-          user_input =f"user name :  {first} \n \n message : أهلا"
-          greet = get_assistant_response(user_input)
-          conversation_history.append({f"AI : {greet}"})
-          return render_template('ai-chat-bot.html' ,account_name=account_name , greet=greet , first=first , last = last , token2 = token2)
-        
     return render_template("sign-in.html", message=message)
 
+
+
 ##############     END LOGIN    #############################################
 
 @app.route('/get_response', methods=['POST'])
@@ -376,8 +407,6 @@ def get_response():
 
     response_text = get_assistant_response(user_input)
     check_image = get_image_Ai(user_input)
-    print (response_text)
-    print (check_image)
 
     if "~" in check_image:
         image_chat.append({f"AI : {check_image}"})