cleanup

app.py

@@ -4,7 +4,6 @@ import requests
 import json
 import os
 
-MODEL = "gpt-4-0125-preview"
 API_URL = os.getenv("API_URL")
 API_KEY = os.getenv("API_KEY")
 
@@ -23,11 +22,9 @@ headers = {
 
 def is_valid_json(data):
     try:
-        # Attempt to parse the JSON data
         parsed_data = json.loads(data)
         return True, parsed_data
     except ValueError as e:
-        # If an error occurs, the JSON is not valid
         return False, str(e)
 
 
@@ -46,7 +43,6 @@ with gr.Blocks() as demo:
     with gr.Row():
 
         with gr.Column(scale=2):
-            # Define inputs for additional parameters
             system_prompt_input = gr.Textbox(
                 label="System Prompt",
                 placeholder="Type system prompt here...",
@@ -86,8 +82,6 @@ with gr.Blocks() as demo:
         global_repetition_penalty = repetition_penalty
 
     def user(user_message, history):
-        # print(f"User: {user_message}")
-        # print(f"History: {history}")
         return "", history + [[user_message, None]]
 
     def bot(
@@ -107,12 +101,7 @@ with gr.Blocks() as demo:
         print(f"Top K: {top_k}")
         print(f"Repetition Penalty: {repetition_penalty}")
 
-        # print(f"History in bot: {history}")
-        # [['Capital of France', 'The capital city of France is Paris.'], ['Thansk', 'You are welcome.'], ['What is the capital of France?', '']]
-        # convert this to [['Capital of France', 'The capital city of France is Paris.'], ['Thansk', 'You are welcome.'], ['What is the capital of France?', '']] to list of dict of role user and assiatant
         history_messages = [{"content": h[0], "role": "user"} for h in history if h[0]]
-        # let's extract the user's question which should be the last touple first element
-        # user_question = history[-1][0]
         history[-1][1] = ""
         sys_msg = [
             {
@@ -140,7 +129,7 @@ with gr.Blocks() as demo:
             },
         }
 
-        # # Making the POST request and streaming the response
+        # Making the POST request and streaming the response
         response = requests.post(
             url, headers=headers, data=json.dumps(data), stream=True
         )
@@ -163,17 +152,11 @@ with gr.Blocks() as demo:
                         )
 
                         if delta_content:  # Ensure there's content to print
-                            # print(f"Bot: {delta_content}")
                             history[-1][1] += delta_content
-                            # print(history)
                             time.sleep(0.05)
                             yield history
                     except json.JSONDecodeError as e:
-                        print(
-                            f"Error decoding JSON: {e} date: {data}"
-                        )  # print(delta_content, flush=True, end="")
-
-                        # print(json_data['choices'][0])
+                        print(f"Error decoding JSON: {e} date: {data}")
 
     msg.submit(
         user, [msg, chatbot], [msg, chatbot], queue=True, concurrency_limit=10