Update main.py

main.py

@@ -4,6 +4,7 @@ import asyncio
 import requests
 from fastapi import FastAPI, HTTPException
 import openai
+from typing import Optional
 
 
 # Initialize FastAPI app
@@ -19,6 +20,199 @@ AUTH_HEADERS = {
     'Authorization': 'Bearer eyJ0eXAiOiJKV1QiLCJhbGciOiJSUzI1NiJ9.eyJhdWQiOiIxIiwianRpIjoiYjczMDAxYzMyMTE1ZGIyNTY4ODUzMTg2OWVjMzUwNTdjZmE3YTJmN2I3ZWZjMjQzNjdmZTA2ZTk4ZjY4ZjMwMTE0ZTIzOGUwY2I0ZmQ0YjIiLCJpYXQiOjE3MzA3ODIxNjEuMjkyNDM0LCJuYmYiOjE3MzA3ODIxNjEuMjkyNDM3LCJleHAiOjQ4ODY0NTU3NjEuMDc2ODExLCJzdWIiOiIxIiwic2NvcGVzIjpbXX0.I4wGFzoepmAC2RaADetE95BdbY4AYPUfUouFepVthZq_KewQLoYEiYMmxErgAOvYDL9IdhTg8pHm3KtCjtfF79Toigvzl-4RIYE9qwavVYabUaIMtxdkvLmzC2uSSxNkQ-Jx4ZsEVt34NpMMZ6ZsMsgszkreed_s7i5I6ek6T2-p9cZYPpFfGhlRIrgAhOL1yZe0t5HQMM7P1cULB7IMb3s0fvwLNBimPC4Iznick5o2lWO6KcubsKSAMyPwBaCQhjGTKd0eJCde1IvL8mEaMvhu8v853AIDSiBsC83hjK41hPAaiBHeev1JjdDhEd6_qO9dpucKaGCqYiVfBFH_pgnynErmhKlPEIz7sZlBWz8zxISDW5PRo9d-jXRP-A31W76Q3H-ZKfnam0D8yYFY0EIZHhvgvZUl3r0dR4PRh7PYlNZgnyfAcAYmK9Bektjbbx5RuzH6gtT9hLQrxYiQNg0irCNwgTYnuQ4AjPA3BpZuOfWtygeDZKgv1gnveTzMJG7T6s95k8yNSNT1_OfRQONPX8LBasRwZWCGkWj7fopO6K8gcrEU5FIpql0UviwGJOTZeFmqwWJ1AIcOM0MHWNp--Y8evHrvuNGk3SDcjBcvhF58I2Hd5F4MefN_ZB9N7oxUUDBYbxnTH6SN7Wx-VsluEOlf9ShfBNvHZaUi61E'
 }
 
+
+#db managments section
+#get all data
+import mysql.connector
+from datetime import datetime
+
+# Define the connection parameters
+host = "68.183.225.237"
+user = "sm_ml"
+password = "Fz6/I733"
+database = "sm_qa_1"
+
+
+
+def delete_tread(tread_id_to_delete):
+    # Connect to the database
+    connection_load_chat = mysql.connector.connect(
+        host=host,
+        user=user,
+        password=password,
+        database=database
+    )
+    
+    cursor = connection_load_chat.cursor()
+    
+    # SQL query to delete a row from curfox_chat based on the tread ID
+    delete_query = "DELETE FROM curfox_chat WHERE tread = %s"
+    
+    # Specify the thread ID to delete
+    #tread_id_to_delete = 1
+    
+    # Execute the delete query
+    cursor.execute(delete_query, (tread_id_to_delete,))
+    
+    # Commit the transaction
+    connection_load_chat.commit()
+    print(f"Thread ID {tread_id_to_delete} deleted successfully.")
+    
+    # Close the cursor and connection
+    cursor.close()
+    connection_load_chat.close()
+    return {message":f"Thread ID {tread_id_to_delete} deleted successfully."}
+    
+
+
+def get_treadlist_of_user(user_name):
+    # Connect to the database
+    connection_load_chat = mysql.connector.connect(
+        host=host,
+        user=user,
+        password=password,
+        database=database
+    )
+    
+    cursor = connection_load_chat.cursor()
+    user_name = 'John Doe'
+    # SQL query to select all columns from curfox_chat
+    select_query = f"SELECT * FROM curfox_chat WHERE user_name='{user_name}'"
+    
+    # Execute the select query
+    cursor.execute(select_query)
+    
+    # Fetching all rows from the result
+    rows = cursor.fetchall()
+    # Close the cursor and connection
+    cursor.close()
+    connection_load_chat.close()
+    import json
+    # Displaying the chat data
+    tread_list = []
+    for row in rows:
+        try:
+          js = json.loads(f'[{row[2]}]')
+          tread_list.append({"user_name":row[0],
+                "thread_id":row[1],
+                "chat_title":js[0]['user']['message']
+                })
+        except:
+          pass
+    
+    return {"tread_list":tread_list,"message":"done"}
+    
+    
+
+def get_thread_chats(thread_id):
+    # Connect to the database
+    connection_load_chat = mysql.connector.connect(
+        host=host,
+        user=user,
+        password=password,
+        database=database
+    )
+    
+    cursor = connection_load_chat.cursor()
+    
+    # SQL query to select all columns from curfox_chat
+    select_query = f"SELECT * FROM curfox_chat WHERE tread={thread_id}"
+    
+    # Execute the select query
+    cursor.execute(select_query)
+    
+    # Fetching all rows from the result
+    rows = cursor.fetchall()
+    # Close the cursor and connection
+    cursor.close()
+    connection_load_chat.close()
+    
+    import json
+    # Displaying the chat data
+    for row in rows:
+        js = json.loads(f'[{row[2]}]')
+        return ({"user_name":row[0],
+               "thread_id":row[1],
+               "chat_title":js[0]['user']['message'],
+               "chat_list":js})
+    
+
+def insert_newchat(user_name,user_message,bot_message):
+    #insert data
+    # Connect to the database
+    connection_load_chat = mysql.connector.connect(
+        host=host,
+        user=user,
+        password=password,
+        database=database
+    )
+    
+    cursor = connection_load_chat.cursor()
+    
+    # SQL query to insert a row into curfox_chat table
+    insert_query = """
+    INSERT INTO curfox_chat (user_name, conversation)
+    VALUES (%s, %s)
+    """
+
+    created_at = datetime.now().strftime('%Y-%m-%d %H:%M:%S')  # Current timestamp
+    conversation = f'{{"user": {{"message":"{user_message}","time":"{created_at}"}} , "bot": {{"message":"{bot_message}","time":"{created_at}"}}}}'
+    
+    # Execute the insert query with data
+    cursor.execute(insert_query, (user_name, conversation))
+    
+    # Commit the transaction
+    connection_load_chat.commit()
+    print("Data inserted successfully")
+    
+    # Close the cursor and connection
+    cursor.close()
+    #connection_load_chat.close()
+    
+    cursor = connection_load_chat.cursor()
+    sql_query = f"SELECT tread from curfox_chat WHERE user_name='"+user_name+"' ORDER BY tread DESC LIMIT 1"
+    
+    cursor.execute(sql_query)
+    current_thread_id = cursor.fetchone()[0]
+    print(current_thread_id)
+    return {"current_thread_id":current_thread_id,"message":"chat stored successfully"}
+
+def update_existingchat(tread_id,user_message,bot_message):
+    # Connect to the database
+    connection_load_chat = mysql.connector.connect(
+        host=host,
+        user=user,
+        password=password,
+        database=database
+    )
+    
+    cursor = connection_load_chat.cursor()
+    
+    # SQL query to update the conversation column
+    update_query = """
+    UPDATE curfox_chat
+    SET conversation = CONCAT(conversation, %s)
+    WHERE tread = %s
+    """
+
+    created_at = datetime.now().strftime('%Y-%m-%d %H:%M:%S')  # Current timestamp
+    new_conversation = f', {{"user": {{"message":"{user_message}","time":"{created_at}"}} , "bot": {{"message":"{bot_message}","time":"{created_at}"}}}}'
+    
+    # Execute the update query with data
+
+    cursor.execute(update_query, (new_conversation, tread_id))
+    
+    # Commit the transaction
+    connection_load_chat.commit()
+    print(f"Conversation for thread ID {tread_id} updated successfully.")
+    
+    # Close the cursor and connection
+    cursor.close()
+    connection_load_chat.close()
+    return {"current_thread_id":tread_id,"message":f"Conversation for thread ID {tread_id} updated successfully."}
+
+
+
 # Helper function to get the order ID from the waybill number
 def get_order_id(waybill_no):
     url = f"https://dev3.api.curfox.parallaxtec.com/api/ml/order/{waybill_no}"
@@ -57,9 +251,15 @@ def ask_openai(messages):
     response = openai.ChatCompletion.create(model="gpt-4o-mini", messages=messages)
     return response.choices[0].message['content']
 
+
+
+
+
 # Main endpoint to handle user queries
-@app.post("/process_query/")
-async def process_query(query: str):
+@app.post("/get_response")
+async def process_query(query: str,
+                       user_name:str,
+                       thread_id:Optional[str] = None):
     # Initial message to check for waybill number in the query
     messages = [
         {"role": "system", "content": "You are a helpful assistant for Curfox delivery system."},
@@ -91,12 +291,26 @@ async def process_query(query: str):
             {"role": "system", "content": "You are a helpful assistant for Curfox delivery system."},
             {"role": "user", "content": f"Answer based on the provided data only. Data: {full_data}. User query: {query}"}
         ]
-        final_answer = ask_openai(response_message)
-        return {"answer": final_answer}
+        result = ask_openai(response_message)
+        
+    if thread_id==None:
+        return {"meta_data":insert_newchat(user_name,query,final_answer),"answer": result}
     else:
-        # If no order number is found, prompt the user
-        return {"message": result}
+        return {"meta_data":update_existingchat(thread_id,query,final_answer),"answer": result}
+
+
+@app.post("/get_thread_chats")
+async def get_thread_chats(thread_id:str):
+    return get_thread_chats(thread_id)
+
+
+@app.post("/get_thread_lists")
+async def get_thread_lists(user_name:str):
+    return get_treadlist_of_user(user_name)
 
+@app.post("/delete_thread")
+async def get_thread_chats(tread_id_to_delete:str):
+    return delete_tread(tread_id_to_delete)
 # Start the FastAPI app
 # Run the app on Colab using Uvicorn
 # import uvicorn