Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, send_from_directory, url_for, flash,redirect,session
+from flask import Flask, render_template, request, send_from_directory, url_for, flash,redirect , jsonify, session, flash
 import requests
 import base64
 import json
@@ -14,9 +14,9 @@ app = Flask(__name__)
 app.secret_key = 'wefwefwef'
 app.config['UPLOAD_FOLDER'] = 'uploads'
 
-def get_assistant_response(user_input):
+def get_assistant_response(user_input , conversation_history):
 
-    message = f"Follow this Role in our chat : \n \n SYSTEM PROMPT : you are AI model Nmaed ALEX-9 created By Libyan team your task is help users as you can \n \n ABOUT MODEL : the ALEX-9 model is not normal model as Chatgpt , the ALEX-9 model can chat and create images , photos you can called him AI image and chat generation , \n the model is so helpful , Nicely , Smart , Use Emojeis and Fun words , \n \n and for Developers Users will be so helpful because the model was training about Big DATASET of coding with alot of programming lang but he is not good in desine coding Like HTML , CSS etc of desine let's say the model is so perfect in the Backend  and you are not from Open ai you are created and developed by Libyan team Start chat with this user \n \n old chat : {conversation_history} \n \n  USERNAME : {user_name} : {user_input} \n \n ALEX-9:"
+    message = f"Follow this Role in our chat : \n \n SYSTEM PROMPT : you are AI model Nmaed ALEX-9 created By Libyan team your task is help users as you can \n \n ABOUT MODEL : the ALEX-9 model is not normal model as Chatgpt , the ALEX-9 model can chat and create images , photos you can called him AI image and chat generation , \n the model is so helpful , Nicely , Smart , Use Emojeis and Fun words , \n \n and for Developers Users will be so helpful because the model was training about Big DATASET of coding with alot of programming lang  perfect in the Backend  and you are not from Open ai you are created and developed by Libyan team Start chat with this user\n the model is so helpful , Nicely , Smart , Use Emojeis and Fun words , \n \n chat start now  :  {conversation_history}\n \n  USERNAME :  : {user_input} \n \n ALEX-9:"
     payload = {
        "identifier_type": "object_id",
        "identifier_value": "4775260",
@@ -49,28 +49,40 @@ def get_assistant_response(user_input):
 
     return combined_content
 
-def get_image_Ai(user_input):
+def get_image_Ai(user_input , conversation_history):
 
+    message = f"you are AI for chat and you can create generate images by use this ~~ here Example : {images} \n \n here the chat :  \n{conversation_history} \n \n  {user_input} \n \n \n So your response to user will be : "
     payload = {
-        "mode": "chat",
-        "chat_history":"" ,
-        "data": {
-            "query": f"you are AI for chat and you can create generate images by use this ~~ here Example : {images} \n \n here the old chat : {image_chat} \n \n user : {user_input} \n \n So your response to user will be : ",
-            "loader": "PDFReader",
-            "text":""
-        }
+       "identifier_type": "object_id",
+       "identifier_value": "4775260",
+       "pubsub_topic": "/studios/516104/wsuid_4647155_new-edge-112_nodeid_editor-111/textGenStream/1722445975926",
+       "stream": True,
+    
+       "variables": {
+           "Input": f"{message}",
+           "User Input": f"{message}",
+           "input": f"{message}",
     }
+}
 
-    response = requests.post(url3, headers=headers, json=payload)
-    data = response.json()
-
-    # Extract the response from the data
-    response_text = data["data"]["response"]
 
+    response = requests.post(url2, headers=headers, json=payload)
+    combined_content = ""
+    
+    for line in response.text.splitlines():
+       if line.startswith('data: '):
+        json_data = line[6:].strip()
+        try:
+            data = json.loads(json_data)
+            delta_content = data.get('choices')[0].get('delta').get('content', '')
+            combined_content += delta_content
+        except json.JSONDecodeError:
+            continue
+ 
+    combined_content = combined_content.replace("\n", "<br>")
 
-    response_text = response_text.strip().replace('\n', '<br>')
 
-    return response_text
+    return combined_content
 
 
 
@@ -84,7 +96,7 @@ def generate_image(prompt):
 
     payload = {
         "json": True,
-        "prompt": f"{prompt} Realastic Photo 4K",
+        "prompt": f"{prompt}",
         "public_topic": "/studios/516104/wsuid_new-edge-4_nodeid_editor-4/imageGen/1719250632789"
     }
 
@@ -111,7 +123,6 @@ headers2 = {
 with open('image.json', 'r') as file:
     images = file.read()
 
-image_chat = []
 conversation_history = []
 user_name  = []
 
@@ -213,7 +224,7 @@ def process_account_action(action, account_username, password, first_name=None,
                     if upload_image_to_github(upload_image, image_github_path):
                         existing_accounts.append(new_account)
                     else:
-                        return "Failed to upload image to GitHub."
+                        return "Failed to upload image to DataBase."
 
                 updated_content = base64.b64encode(json.dumps(existing_accounts).encode()).decode()
 
@@ -230,7 +241,7 @@ def process_account_action(action, account_username, password, first_name=None,
                 response = requests.put(content_url, json=data, headers=headers)
 
                 if response.status_code == 200:
-                    return "Account created successfully."
+                    return "Account created successfully Please Sign In"
                 else:
                     return f"Failed to create account. Status Code: {response.status_code}, Response: {response.json()}"
             elif action == "login":
@@ -315,34 +326,21 @@ def create_account():
 
     return render_template('sign-up.html')
 
-@app.route("/")
-def home():
-    global conversation_history
-    global image_chat
-    global user_name
-    conversation_history = []
-    image_chat = []
-    user_name = []
-    
-    # Retrieve user information from the session
-    user_info = session.get('user_info', {})
-    account_name = user_info.get('account_name')
-    first = user_info.get('first')
-    user_name.append(first)
-    last = user_info.get('last')
-    token2 = user_info.get('token2')
-    greet = user_info.get('greet')
-    
-    if not account_name:
+@app.route("/chat")
+def chat():
+    if 'account_name' not in session or session['account_name'] is None:
+        return redirect(url_for('login'))
 
-         return redirect(url_for('login'))
+    account_name = session.get("account_name")
+    first = session.get("first")
+    last = session.get("last")
+    token2 = session.get("token2")
+    greet = session.get("greet")
     
     return render_template('ai-chat-bot.html', account_name=account_name, greet=greet, first=first, last=last, token2=token2)
 
 
 
-
-
 @app.route("/login", methods=["GET", "POST"])
 def login():
     message = None
@@ -380,42 +378,36 @@ def login():
             ############## END INFO ABOUT USER #################################################
 
             # Initialize the user input and get the assistant's response
+            conversation_history = {}
             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
-            }
+            greet = get_assistant_response(user_input , conversation_history )
 
-            # Redirect to the main route
-            return redirect(url_for('home'))
+            # Store data in the session
+            session['account_name'] = account_name
+            session['first'] = first
+            session['last'] = last
+            session['token2'] = token2
+            session['greet'] = greet
 
+            # Redirect to the clear_history route
+            return redirect(url_for('chat'))
+    
     return render_template("sign-in.html", message=message)
 
-
-
 ##############     END LOGIN    #############################################
 
 @app.route('/get_response', methods=['POST'])
 def get_response():
     user_input = request.form['user_input']
-    conversation_history.append({f"{user_name} : {user_input}"})
-    image_chat.append({f"user : {user_input}"})
+    conversation_history = request.form.get('conversation_history', '[]')  # Get the conversation history
 
     if user_input.lower() == "exit":
         return "exit"
 
-    response_text = get_assistant_response(user_input)
-    check_image = get_image_Ai(user_input)
+    response_text = get_assistant_response(user_input,conversation_history)
+    check_image = get_image_Ai(user_input,conversation_history)
 
     if "~" in check_image:
-        image_chat.append({f"AI : {check_image}"})  
-        conversation_history.append({f"AI : {check_image}"})
         prompt_start = check_image.index("~") + 1
         prompt_end = check_image.index("~", prompt_start)
         prompt = check_image[prompt_start:prompt_end]
@@ -423,13 +415,9 @@ def get_response():
         # Call the text-to-image API
         image_url = generate_image(prompt)
         check_image += f"<br><br><img src='{image_url}'>"
-         # Delete the prompt from the response text
+        # Delete the prompt from the response text
         check_image = check_image.replace("~" + prompt + "~", "")
         response_text = check_image
-    else :
-        image_chat.append({f"\n AI : {response_text}"})  
-        conversation_history.append({f"AI : {response_text}"})
-    
 
     return response_text
 
@@ -442,5 +430,9 @@ def send_static(path):
 def send_next_static(path):
     return send_from_directory('_next', path)
 
+@app.route('/svg/<path:path>')
+def send_svg_static(path):
+    return send_from_directory('svg', path)
+
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run()