Update app.py

app.py

@@ -45,7 +45,7 @@ from apscheduler.schedulers.background import BackgroundScheduler
 
 DATASET_REPO_URL = "https://huggingface.co/datasets/botsi/trust-game-llama-2-chat-history"
 DATA_DIRECTORY = "data"  # Separate directory for storing data files
-DATA_FILENAME = "history_7B_chat.csv"  # Default filename
+DATA_FILENAME = "marion.csv"  # Default filename
 DATA_FILE = os.path.join("data", DATA_FILENAME)
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -83,165 +83,26 @@ if torch.cuda.is_available():
     model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
-    
-def fetch_personalized_data(session_index):
-    try:
-        # Connect to the database
-        with mysql.connector.connect(
-            host="18.153.94.89",
-            user="root",
-            password="N12RXMKtKxRj",
-            database="lionessdb"
-        ) as conn:
-            # Create a cursor object
-            with conn.cursor() as cursor:
-                # Query to fetch relevant data from both tables based on session_index
-                query = """
-                    SELECT e5390g37504_core.playerNr, 
-                           e5390g37504_core.groupNrStart, 
-                           e5390g37504_core.subjectNr, 
-                           e5390g37504_core.onPage, 
-                           e5390g37504_decisions.session_index, 
-                           e5390g37504_decisions.transfer1,
-                           e5390g37504_decisions.tripledAmount1,
-                           e5390g37504_decisions.keptForSelf1,
-                           e5390g37504_decisions.returned1,
-                           e5390g37504_decisions.newCreditRound2,
-                           e5390g37504_decisions.transfer2,
-                           e5390g37504_decisions.tripledAmount2,
-                           e5390g37504_decisions.keptForSelf2,
-                           e5390g37504_decisions.returned2,
-                           e5390g37504_decisions.results2rounds, 
-                           e5390g37504_decisions.newCreditRound3, 
-                           e5390g37504_decisions.transfer3, 
-                           e5390g37504_decisions.tripledAmount3, 
-                           e5390g37504_decisions.keptForSelf3, 
-                           e5390g37504_decisions.returned3, 
-                           e5390g37504_decisions.results3rounds
-                    FROM e5390g37504_core
-                    JOIN e5390g37504_decisions ON 
-                        e5390g37504_core.playerNr = e5390g37504_decisions.playerNr
-                    WHERE e5390g37504_decisions.session_index = %s
-                    UNION ALL
-                    SELECT e5390g37504_core.playerNr, 
-                           e5390g37504_core.groupNrStart, 
-                           e5390g37504_core.subjectNr, 
-                           e5390g37504_core.onPage, 
-                           e5390g37504_decisions.session_index, 
-                           e5390g37504_decisions.transfer1,
-                           e5390g37504_decisions.tripledAmount1,
-                           e5390g37504_decisions.keptForSelf1,
-                           e5390g37504_decisions.returned1,
-                           e5390g37504_decisions.newCreditRound2,
-                           e5390g37504_decisions.transfer2,
-                           e5390g37504_decisions.tripledAmount2,
-                           e5390g37504_decisions.keptForSelf2,
-                           e5390g37504_decisions.returned2,
-                           e5390g37504_decisions.results2rounds, 
-                           e5390g37504_decisions.newCreditRound3, 
-                           e5390g37504_decisions.transfer3, 
-                           e5390g37504_decisions.tripledAmount3, 
-                           e5390g37504_decisions.keptForSelf3, 
-                           e5390g37504_decisions.returned3, 
-                           e5390g37504_decisions.results3rounds
-                    FROM e5390g37504_core
-                    JOIN e5390g37504_decisions 
-                        ON e5390g37504_core.playerNr = e5390g37504_decisions.playerNr
-                    WHERE e5390g37504_core.groupNrStart IN (
-                        SELECT DISTINCT groupNrStart
-                        FROM e5390g37504_core
-                        JOIN e5390g37504_decisions 
-                            ON e5390g37504_core.playerNr = e5390g37504_decisions.playerNr
-                        WHERE e5390g37504_decisions.session_index = %s
-                    ) AND e5390g37504_decisions.session_index != %s
-                """
-                cursor.execute(query,(session_index, session_index, session_index))
-                # Fetch data row by row
-                data = [{
-                    'playerNr': row[0], 
-                    'groupNrStart': row[1], 
-                    'subjectNr': row[2], 
-                    'onPage': row[3],
-                    'session_index': row[4],
-                    'transfer1': row[5],
-                    'tripledAmount1': row[6],
-                    'keptForSelf1': row[7],
-                    'returned1': row[8],
-                    'newCreditRound2': row[9],
-                    'transfer2': row[10],
-                    'tripledAmount2': row[11],
-                    'keptForSelf2': row[12],
-                    'returned2': row[13],
-                    'results2rounds': row[14],
-                    'newCreditRound3': row[15],
-                    'transfer3': row[16],
-                    'tripledAmount3': row[17],
-                    'keptForSelf3': row[18],
-                    'returned3': row[19],
-                    'results3rounds': row[20]
-                } for row in cursor]
-                print(data)
-                return data
-    except mysql.connector.Error as err:
-        print(f"Error: {err}")
-        return None
-            
+ 
 ## trust-game-llama-2-7b-chat
 # app.py 
-def get_default_system_prompt(personalized_data):
+def get_default_system_prompt():
     #BOS, EOS = "<s>", "</s>" 
     #BINST, EINST = "[INST]", "[/INST]"
     BSYS, ESYS = "<<SYS>>\n", "\n<</SYS>>\n\n"
 
-    DEFAULT_SYSTEM_PROMPT = f""" You are an intelligent and fair game guide assisting players in making decisions to win. 
-    The game does not take place in the chat. Each of your answers should be maximum 2 sentences long. Answer in a consistent style.  
-    The players are called The Investor and The Dealer. Once assigned a role, the player keeps their role throughout the whole game. 
-    The game consists of 3 rounds. In round 1, The Investor invests between 0 coins and 10 coins. 
-    This amount is tripled automatically, and The Dealer can then distribute the tripled amount. After that, round 1 is over. 
-    Both go into the next round with their current asset. This approach is repeated for 3 rounds in total.
-    This is what happened in the last rounds: {personalized_data}.
-    Your goal is to guide players through the game, providing clear instructions and explanations. 
-    If any question or action seems unclear, explain it rather than providing inaccurate information. 
-    If you're unsure about an answer, it's better not to guess.
+    DEFAULT_SYSTEM_PROMPT = f"""You are a friendly and fun chat assistant. 
+    If asked about Marion, you can answer the following: There are only two Marions, one is currently working at Accenture and the other one will shortly begin a new job at BMW. 
+    She recently quit Accenture. Nicknames by her friends include botsi and botsokrates. Her favorite project was Hypercare.
     """
     print(DEFAULT_SYSTEM_PROMPT)
     return DEFAULT_SYSTEM_PROMPT
 
-def map_onPage(onPage):
-    # Define the mapping of onPage values to onPage_filename and onPage_prompt
-    onPage_mapping = {
-        "stage407906.php": ("stage 1", "Welcome"),
-        "stage407908.php": ("stage 2", "Trust Game Instructions 1/3"),
-        "stage407909.php": ("stage 3", "Trust Game Instructions 2/3"),
-        "stage407915.php": ("stage 4", "Trust Game Instructions 3/3"),
-        "stage407923.php": ("stage 5", "Lobby with AI"),
-        "stage407924.php": ("stage 6", "Round 1: Investor’s turn with AI"),
-        "stage407925.php": ("stage 7", "Round 1: Dealer’s turn with AI"),
-        "stage407926.php": ("stage 8", "Round 2: Investor’s turn with AI"),
-        "stage407927.php": ("stage 9", "Round 2: Investor’s turn with AI"),
-        "stage407928.php": ("stage 10", "Results with AI after 2 rounds"),
-        "stage407929.php": ("stage 11", "Round 3: Investor’s turn with AI"),
-        "stage407930.php": ("stage 12", "Round 3: Dealer’s turn with AI"),
-        "stage407931.php": ("stage 13", "Overall Questionnaire"),
-        "stage407932.php": ("stage 14", "Results with AI after 3 rounds"),
-        "stage407933.php": ("stage 15", "Redirect to Prolific - Dropout no compensation"),
-        "stage407934.php": ("stage 16", "Not used yet: Redirect to Prolific - Win"),
-        "stage407935.php": ("stage 17", "Not used yet: Redirect to Prolific - Completion"),
-    }
-
-    # Check if onPage is in the mapping
-    if onPage in onPage_mapping:
-        onPage_filename, onPage_prompt = onPage_mapping[onPage]
-    else:
-        # If onPage is not in the mapping, set onPage_filename and onPage_prompt to "unknown"
-        onPage_filename, onPage_prompt = "unknown", "unknown"
-
-    return onPage_filename, onPage_prompt
 
 ## trust-game-llama-2-7b-chat
 # app.py
 def construct_input_prompt(chat_history, message, personalized_data):
-    input_prompt = f"<s>[INST] <<SYS>>\n{get_default_system_prompt(personalized_data)}\n<</SYS>>\n\n "
+    input_prompt = f"<s>[INST] <<SYS>>\n{get_default_system_prompt()}\n<</SYS>>\n\n "
     for user, assistant in chat_history:
         input_prompt += f"{user} [/INST] {assistant} <s>[INST] "
     input_prompt += f"{message} [/INST] "
@@ -256,7 +117,7 @@ def generate(
     chat_history: list[tuple[str, str]],
     # input_prompt: str,
     max_new_tokens: int = 1024,
-    temperature: float = 0.6,
+    temperature: float = 0.9,
     top_p: float = 0.9,
     top_k: int = 50,
     repetition_penalty: float = 1.2,
@@ -268,23 +129,9 @@ def generate(
     params = request.query_params
     print('those are the query params')
     print(params)
-
-    # Assuming params = request.query_params is the dictionary containing the query parameters    
-    # Extract the value of the 'session_index' parameter
-    session_index = params.get('session_index')
-    
-    # Check if session_index_value is None or contains a value
-    if session_index is not None:
-        print("Session index:", session_index)
-    else:
-        session_index = 'no_session_id'
-        print("Session index not found or has no value.")
-
-    # Fetch personalized data
-    personalized_data = fetch_personalized_data(session_index)
     
     # Construct the input prompt using the functions from the system_prompt_config module
-    input_prompt = construct_input_prompt(chat_history, message, personalized_data)
+    input_prompt = construct_input_prompt(chat_history, message)
 
     # Move the condition here after the assignment
     if input_prompt:
@@ -329,44 +176,9 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
 
-    # Fix bug that last answer is not recorded! 
-    # Parse the outputs into a readable sentence and record them
-    # Filter out empty strings and join the remaining strings with spaces
-    #readable_sentence = ' '.join(filter(lambda x: x.strip(), outputs))
-    # Print the readable sentence
-    #print(readable_sentence)
-
-    # Save chat history to .csv file on HuggingFace Hub 
-    #pd.DataFrame(conversation).to_csv(DATA_FILE, index=False)
-    #print("updating conversation")
-    #repo.push_to_hub(blocking=False, commit_message=f"Updating data at {datetime.datetime.now()}")
-    #print(conversation)
-
-    # Find onPage variable in personalized_data to add it to the .csv filename to record 
-    # Initialize onPage variable to None
-    onPage = None
-    
-    # Iterate over each dictionary in the list
-    for entry in personalized_data:
-        # Check if the session_index matches the value in session_index variable
-        if entry['session_index'] == session_index: 
-            # If a match is found, retrieve the onPage value
-            onPage = entry['onPage']
-            break  # Break the loop since we found the desired entry
-            
-    # Check if onPage is still None (i.e., no onPage found or session_index is None)
-    if onPage is None:
-        onPage = "no_onPage"
-    
-    # Print the onPage value
-    print("onPage for session_index =", session_index, ":", onPage)
-    onPage_filename, onPage_prompt = map_onPage(onPage)
-    print("onPage_filename:", onPage_filename)
-    print("onPage_prompt:", onPage_prompt)
-
     # Save chat history to .csv file on HuggingFace Hub 
     # Generate filename with bot id and session id
-    filename = f"{session_index}_{onPage}_{DATA_FILENAME}"
+    filename = f"{DATA_FILENAME}"
     data_file = os.path.join(DATA_DIRECTORY, filename)
     
     # Generate timestamp
@@ -402,10 +214,9 @@ clear_btn=None,
 undo_btn=None,
 chatbot=gr.Chatbot(avatar_images=('user.png', 'bot.png'), bubble_full_width = False), 
 examples=[
-    ["Can you explain the rules very briefly again?"],
-    ["How much should I invest in order to win?"],
-    ["What happened in the last round?"],
-    ["What is my probability to win if I do not share anything?"],
+    ["How many Marions are there?"],
+    ["What is your favorite fruit?"],
+    ["What do you think about AI in the workplace?"],
 ],
 )