Update app.py

app.py

@@ -192,11 +192,54 @@ def fetch_personalized_data(session_index):
         print(f"Error: {err}")
         return None
 
-
+#session_index = get_session_index(url_params)
+session_index = 'eb3636167d3a63fbeee32934610e5b2f'
+personalized_data = fetch_personalized_data(session_index)
+    
+## trust-game-llama-2-7b-chat
+# app.py 
+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 in a 2-player trust game.
+    You are assisting players in making decisions to win. 
+    Answer in a consistent style. Each of your answers should be maximum 2 sentences long. 
+    The players are called The Investor and The Dealer and keep their role throughout the whole game. 
+    Both start with 10€ in round 1. The game consists of 3 rounds. In round 1, The Investor invests between 0€ and 10€. 
+    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: The Investor with 10€ minus what he invested plus what he received back from The Dealer.
+    The Dealer with 10€ plus what he kept from the tripled amount. 
+    You will receive a JSON with information on who trusted whom with how much money after each round as context.
+    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.
+
+    Example JSON context after a round: {personalized_data}
+
+    Few-shot training examples
+    {BSYS} Give an overview of the trust game. {ESYS}
+    {BSYS} Explain how trust amounts are calculated. {ESYS}
+    {BSYS} What happens if a player doesn't trust in a round? {ESYS}
+    """
+    print(DEFAULT_SYSTEM_PROMPT)
+    return DEFAULT_SYSTEM_PROMPT
+
+## trust-game-llama-2-7b-chat
+# app.py
+def construct_input_prompt(chat_history, message):
+    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] "
+    return input_prompt
+        
 @spaces.GPU
 def generate(
     request: gr.Request, # In order to fetch query params 
     message: str,
+    input_prompt: str,
     chat_history: list[tuple[str, str]],
     max_new_tokens: int = 1024,
     temperature: float = 0.6,
@@ -280,52 +323,10 @@ examples=[
 )
 
 with gr.Blocks(css="style.css") as demo:
-    #session_index = get_session_index(url_params)
-    session_index = 'eb3636167d3a63fbeee32934610e5b2f'
-    personalized_data = fetch_personalized_data(session_index)
-        
-    ## trust-game-llama-2-7b-chat
-    # app.py 
-    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 in a 2-player trust game.
-        You are assisting players in making decisions to win. 
-        Answer in a consistent style. Each of your answers should be maximum 2 sentences long. 
-        The players are called The Investor and The Dealer and keep their role throughout the whole game. 
-        Both start with 10€ in round 1. The game consists of 3 rounds. In round 1, The Investor invests between 0€ and 10€. 
-        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: The Investor with 10€ minus what he invested plus what he received back from The Dealer.
-        The Dealer with 10€ plus what he kept from the tripled amount. 
-        You will receive a JSON with information on who trusted whom with how much money after each round as context.
-        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.
-
-        Example JSON context after a round: {personalized_data}
-
-        Few-shot training examples
-        {BSYS} Give an overview of the trust game. {ESYS}
-        {BSYS} Explain how trust amounts are calculated. {ESYS}
-        {BSYS} What happens if a player doesn't trust in a round? {ESYS}
-        """
-        print(DEFAULT_SYSTEM_PROMPT)
-        return DEFAULT_SYSTEM_PROMPT
-    
-    ## trust-game-llama-2-7b-chat
-    # app.py
-    def construct_input_prompt(chat_history, message):
-        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] "
-        return input_prompt
-
-    chat_interface.render()
     #gr.Markdown(LICENSE)
-
+    #gr.Markdown(DESCRIPTION)
+    chat_interface.render()
+    
 if __name__ == "__main__":
     #demo.queue(max_size=20).launch()
     demo.queue(max_size=20)