Update app.py

app.py

@@ -1,4 +1,6 @@
-# Original code from https://huggingface.co/spaces/huggingface-projects/llama-2-7b-chat and https://huggingface.co/spaces/radames/gradio-chatbot-read-query-param 
+# Original code from https://huggingface.co/spaces/huggingface-projects/llama-2-7b-chat
+# Modified for trust game purposes
+
 import gradio as gr
 import time
 import random
@@ -6,22 +8,21 @@ import json
 import mysql.connector
 import os
 import csv
-from huggingface_hub import Repository, hf_hub_download
-from datetime import datetime
-
 import spaces
 import torch
+
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from threading import Thread
 from typing import Iterator
+from huggingface_hub import Repository, hf_hub_download
+from datetime import datetime
 
-# data_fetcher.py
+# for fetch_personalized_data
 import mysql.connector
 import urllib.parse
 import urllib.request
 
-
-# Save chat history as JSON
+# for saving chat history as JSON
 import atexit
 import os
 from huggingface_hub import HfApi, HfFolder
@@ -69,10 +70,10 @@ DESCRIPTION = """\
 This is your personal space to chat. 
 You can ask anything from strategic questions regarding the game or just chat as you like. 
 """
+
+# LICENSE commented out 
 '''LICENSE = """
 <p/>
-
----
 As a derivate work of [Llama-2-13b-chat](https://huggingface.co/meta-llama/Llama-2-13b-chat) by Meta,
 this demo is governed by the original [license](https://huggingface.co/spaces/huggingface-projects/llama-2-13b-chat/blob/main/LICENSE.txt) and [acceptable use policy](https://huggingface.co/spaces/huggingface-projects/llama-2-13b-chat/blob/main/USE_POLICY.md).
 """
@@ -88,7 +89,6 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
 
-import mysql.connector
 
 def fetch_personalized_data(session_index):
     try:
@@ -193,31 +193,11 @@ def fetch_personalized_data(session_index):
         return None
 
 
-js = """
-    function() {
-        const params = new URLSearchParams(window.location.search);
-        const url_params = Object.fromEntries(params);
-        return url_params;
-        }
-    """
-
-def get_window_url_params():
-    return """
-        function() {
-            const params = new URLSearchParams(window.location.search);
-            const url_params = Object.fromEntries(params);
-            return url_params;
-            }
-        """
-
-## trust-game-llama-2-7b-chat
-# app.py 
 @spaces.GPU
 def generate(
-    request: gr.Request, 
+    request: gr.Request, # In order to fetch query params 
     message: str,
     chat_history: list[tuple[str, str]],
-    # system_prompt: str,
     max_new_tokens: int = 1024,
     temperature: float = 0.6,
     top_p: float = 0.9,
@@ -225,14 +205,12 @@ def generate(
     repetition_penalty: float = 1.2,
 ) -> Iterator[str]: # Change return type hint to Iterator[str]
 
+    # Fetch query params 
     params = request.query_params
-    print(params)
+    print('those are the query params:' + params)
 
-    # Construct the input prompt using the functions from the system_prompt_config module
+    # Construct the input prompt using the functions from the construct_input_prompt function 
     input_prompt = construct_input_prompt(chat_history, message)
-
-    # Use the global variable to store the chat history
-    # global global_chat_history
     
     conversation = []
 
@@ -300,8 +278,7 @@ examples=[
 ],
 )
 
-with gr.Blocks(js = js, css="style.css") as demo:
-    #url_params = gr.JSON({}, visible=False, label="URL Params")
+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)