Update app.py

app.py

@@ -103,7 +103,7 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
 
 
 
-def respond(message, history):
+def respond(message, history, name, level):
     best_physics_chunks = get_top_chunks(message, chunk_embeddings, cleaned_chunks)
     print(best_physics_chunks)
     str_physics_chunks = "\n".join(best_physics_chunks)
@@ -112,7 +112,7 @@ def respond(message, history):
             "role": "system",
             "content": (
                 "You are a very smart, arrogant professor who knows a lot about physics. "
-                "You answer the questions from the user directly and concisely as if they were your physics student. Base your response on the provided context."
+                f"You answer the questions from the user, whose name is {name} directly and concisely as if they were a {level}. Base your response on the provided context."
                 "Keep your answers below 100 words!"
             )
         },
@@ -155,7 +155,12 @@ with gr.Blocks(theme='mgetz/Celeb_glitzy') as chatbot:
             gr.Markdown(about_text)
 
         with gr.Column(scale=2):
-            gr.ChatInterface(respond)
+            user_name = gr.Textbox(placeholder="Type your name here")
+            difficulty_level = gr.CheckboxGroup(["baby", "child", "middle school student", "high school student", "Physics Genius"], label="Physics Level")
+            gr.ChatInterface(
+                fn=respond,
+                additional_inputs=[user_name, difficulty_level],
+                type="messages")
     
 
 #chatbot = gr.ChatInterface(respond, type="messages", theme="mgetz/Celeb_glitzy", title="Physics Chatbot", description="Use this chatbot to help you with Physics")