Update app.py

app.py

@@ -91,21 +91,18 @@ Try to be funny, use smart format to answer using bullet points.
 Begin the tutorial by introducing yourself and asking for the first prompt as described above.
 """
 
-if __name__ == '__main__':
-    api_token = os.getenv('HF_API_TOKEN2')
-    if not api_token:
-        raise ValueError("HF_API_TOKEN not found in environment variables")
-        
+api_token = os.getenv('HF_API_TOKEN2')
+if not api_token:
+    raise ValueError("HF_API_TOKEN not found in environment variables")
+
 client = InferenceClient("meta-llama/Meta-Llama-3-70B-Instruct", token=api_token)
 
-def respond(message, history: list[tuple[str, str]]):
+def respond(message, chat_history):
     messages = [{"role": "system", "content": system_message}]
 
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    for user_msg, assistant_msg in chat_history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
 
     messages.append({"role": "user", "content": message})
 
@@ -119,12 +116,12 @@ def respond(message, history: list[tuple[str, str]]):
         top_p=0.95,
     ):
         token = message.choices[0].delta.content
-
         response += token
         yield response
 
 demo = gr.ChatInterface(
     respond,
+    chatbot=gr.Chatbot(scale=1, placeholder=PLACEHOLDER),
     theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue", neutral_hue="gray",font=[gr.themes.GoogleFont("Exo"), "ui-sans-serif", "system-ui", "sans-serif"]).set(
         body_background_fill_dark="#0f172a",
         block_background_fill_dark="#0f172a",
@@ -136,15 +133,9 @@ demo = gr.ChatInterface(
         color_accent_soft_dark="transparent"
     ),
     css=css,
-    description="AI Prompt Engineering Tutor: Master the art of crafting effective prompts",
-    chatbot=gr.Chatbot(scale=1, placeholder=PLACEHOLDER),
-    additional_inputs=[
-        gr.Button("Retry", variant="secondary"),
-        gr.Button("Undo", variant="secondary"),
-        gr.Button("Clear", variant="secondary"),
-    ],
-    submit_btn=gr.Button("Send", variant="primary"),
+    title="AI Prompt Engineering Tutor",
+    description="Master the art of crafting effective prompts",
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)