Update app.py

app.py

@@ -10,7 +10,8 @@ from googleapiclient.http import MediaFileUpload
 from pathlib import Path
 import argparse
 from huggingface_hub import snapshot_download
-
+import random 
+import string
 
 #TODO : Cleanup and comment
 
@@ -56,26 +57,26 @@ Occupation = ""
 Ethnicity = ""
 Gender = ""
 Age=""
+YearsOfExp = ""
 
 chat_log_name =""
 
 from llama_cpp import Llama
 llm = Llama(model_path=model_file, model_type="mistral",n_gpu_layers=-1,n_ctx = 2048)
 
+unique_id = generate_unique_id()
 
 def load_model():
     global llm
     llm = Llama(model_path=model_file, model_type="mistral",n_gpu_layers=-1,n_ctx = 2048)
     return "Model loaded"
 
-def reset_chat():
-    global chat_history, chat_log_history, isFirstRun, context, chat_log_name
-    chat_history = []
-    chat_log_history = []
-    isFirstRun = True
-    context = initContext  # Assuming initContext is your initial context
-    chat_log_name = ""
-    return "Chat reset successfully. You can start a new chat now."
+def generate_unique_id():
+    # Generate a random sequence of 3 letters and 3 digits
+    letters = ''.join(random.choices(string.ascii_letters, k=3))
+    digits = ''.join(random.choices(string.digits, k=3))
+    unique_id = letters + digits
+    return unique_id
 
 def get_drive_service():
     credentials = service_account.Credentials.from_service_account_file(
@@ -116,6 +117,7 @@ def upload_to_google_drive():
     data = {
             "name": Name,
             "occupation": Occupation,
+            "years of experience": YearsOfExp,
             "ethnicity": Ethnicity,
             "gender": Gender,
             "age": Age,
@@ -145,10 +147,9 @@ def upload_to_google_drive():
 
 def generate(prompt, history):
 
-    global isFirstRun,initContext,Name,Occupation,Ethnicity,Gender,Age,context
+    global isFirstRun,initContext,Name,Occupation,Ethnicity,Gender,Age,context,YearsOfExp
     
-    if not len(Name) == 0 and not len(Occupation) == 0 and not len(Ethnicity) == 0 and not len(Gender) == 0 and not len(Age) == 0:
-
+    if not len(Name) == 0 and not len(Occupation) == 0 and not len(Ethnicity) == 0 and not len(Gender) == 0 and not len(Age) == 0 and not len(YearsOfExp):
         firstmsg =""
         if isFirstRun:
             context = initContext
@@ -197,16 +198,17 @@ def generate(prompt, history):
         yield output
 
 
-def name_interface(name,occupation,ethnicity,gender,age,reset_button):
-    global Name, Occupation,Ethnicity,Gender,Age,chat_log_name
+def name_interface(name,occupation,yearsofexp,ethnicity,gender,age,reset_button):
+    global Name, Occupation,Ethnicity,Gender,Age,chat_log_name,YearsOfExp
     print(reset_button)
     Name = name
     Occupation = occupation
     Ethnicity=ethnicity
     Gender=gender
     Age=age
+    YearsOfExp = yearsofexp
     
-    if name and occupation and ethnicity and gender and age: 
+    if name and occupation and ethnicity and gender and age and yearsofexp: 
         chat_log_name =  f'chat_log_for_{Name}_{datetime.now().strftime("%Y-%m-%d_%H-%M-%S")}.json'
         return f"You can start chatting now {Name}"    
     else:
@@ -220,9 +222,10 @@ def reset_chat_interface():
     return "Chat has been reset."
 
 def reset_name_interface():
-    global Name, Occupation, Ethnicity, Gender, Age, chat_log_name
+    global Name, Occupation, Ethnicity, Gender, Age,YearsOfExp, chat_log_name
     Name = ""
     Occupation = ""
+    YearsOfExp = ""
     Ethnicity = ""
     Gender = ""
     Age = ""
@@ -230,10 +233,12 @@ def reset_name_interface():
     return "User info has been reset."
 
 def reset_all():
+    global unique_id
     message1 = reset_chat_interface()
     message2 = reset_name_interface()
     message3 = load_model()
-    return f"{message1} || {message2} || {message3}"
+    unique_id = generate_unique_id()
+    return f"All Chat components have been rest.  Uniqe ID for this session is, {unique_id}. Please note this down."
 
 chat_bot=gr.ChatInterface(
     fn=generate,
@@ -244,10 +249,12 @@ chat_bot=gr.ChatInterface(
 name_interface = gr.Interface(
     fn=name_interface,
     inputs=[
+        gr.Textbox(label="Unique ID", value = unique_id ,container = True, scale = 2 ,interactive = False, show_copy_button = True),
         gr.Textbox(label="Name", placeholder="Enter your name here..."),
         gr.Textbox(label="Occupation", placeholder="Enter your occupation here..."),
+        gr.Textbox(label="Years of Experience", placeholder="Enter the years of experience"),
         gr.Textbox(label="Ethnicity", placeholder="Enter your Ethnicity here..."),
-        gr.Dropdown(choices=["Male", "Female", "Other"], label="Gender"),
+        gr.Dropdown(choices=["Male", "Female","Other","Prefer Not To Say"], label="Gender"),
         gr.Textbox(label="Age", placeholder="Enter your Age here..."),
     ],outputs="text",
     title="ECU-IVADE : User Information",