Update app.py

app.py

@@ -2,16 +2,18 @@ import os
 import gradio as gr
 from text_generation import Client, InferenceAPIClient
 
+#variable contains a pre-prompt string for the chat interaction.
 openchat_preprompt = (
     "\n<human>: Zdravo!\n<bot>: \n"
 )
 
+#This function returns a client object based on the provided model.
 def get_client(model: str):
     if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
         return Client(os.getenv("OPENCHAT_API_URL"))
     return InferenceAPIClient(model, token=os.getenv("HF_TOKEN", None))
 
-
+#This function returns pre-prompt, username, bot name, and separator strings based on the provided model.
 def get_usernames(model: str):
     """
     Returns:
@@ -23,7 +25,7 @@ def get_usernames(model: str):
         return openchat_preprompt, "<human>: ", "<bot>: ", "\n"
     return "", "User: ", "Assistant: ", "\n"
 
-
+#This function performs the text generation prediction based on the provided inputs and model parameters.
 def predict(
     model: str,
     inputs: str,
@@ -101,11 +103,11 @@ def predict(
         ]
         yield chat, history
 
-
+#This function resets the input textbox of the app.
 def reset_textbox():
     return gr.update(value="")
 
-
+#This function handles the change in the selected model radio button and updates the visibility and values of other related input elements accordingly.
 def radio_on_change(
     value: str,
     disclaimer,
@@ -161,6 +163,7 @@ text_generation_inference = """
 openchat_disclaimer = """
 """
 
+#This sets up the Gradio app interface using the gr.Blocks context manager.
 with gr.Blocks(
     css="""#col_container {margin-left: auto; margin-right: auto;}
                 #chatbot {height: 520px; overflow: auto;}"""
@@ -233,7 +236,8 @@ with gr.Blocks(
                 visible=False,
             )
             watermark = gr.Checkbox(value=False, label="Vodeni žig teksta")
-
+            
+    #method is used to trigger actions when the selected model radio button changes.
     model.change(
         lambda value: radio_on_change(
             value,
@@ -257,6 +261,7 @@ with gr.Blocks(
         ],
     )
 
+    #methods define actions to be performed when the input textbox is submitted or a button is clicked.
     inputs.submit(
         predict,
         [
@@ -291,6 +296,8 @@ with gr.Blocks(
     )
     b1.click(reset_textbox, [], [inputs])
     inputs.submit(reset_textbox, [], [inputs])
-
+    
+    #The app description markdown is displayed. The app is launched with the specified concurrency count and debug mode enabled.
     gr.Markdown(description)
-    demo.queue(concurrency_count=16).launch(debug=True)
+    demo.queue(concurrency_count=16).launch(debug=True)
+