Add adaptive UI to loading state and switch model

app.py

@@ -7,13 +7,14 @@ from dotenv import load_dotenv
 load_dotenv()
 
 # Initialize Cohere API client
-co = cohere.Client()
+co = cohere.Client(os.getenv('COHERE_API_KEY'))
+MODEL_NAME = 'command-r-plus'  # Define model name as a constant
 
 # Adaptive learning functions
 def assess_knowledge(name, experience, goals):
     try:
         level_prompt = f"User Name: {name}, Experience: {experience}, Goals: {goals}. Classify as beginner, intermediate, or advanced."
-        response = co.generate(prompt=level_prompt)
+        response = co.generate(prompt=level_prompt, model=MODEL_NAME)
         level = response.generations[0].text.strip()
         return level
     except Exception as e:
@@ -22,7 +23,7 @@ def assess_knowledge(name, experience, goals):
 def generate_explanation(topic, level):
     try:
         explanation_prompt = f"Explain the topic '{topic}' to a {level} level student."
-        response = co.generate(prompt=explanation_prompt)
+        response = co.generate(prompt=explanation_prompt, model=MODEL_NAME)
         explanation = response.generations[0].text.strip()
         return explanation
     except Exception as e:
@@ -31,7 +32,7 @@ def generate_explanation(topic, level):
 def generate_challenge(topic, level):
     try:
         challenge_prompt = f"Generate a {level} level challenge for learning '{topic}' in computer science."
-        response = co.generate(prompt=challenge_prompt)
+        response = co.generate(prompt=challenge_prompt, model=MODEL_NAME)
         challenge = response.generations[0].text.strip()
         return challenge
     except Exception as e:
@@ -80,12 +81,26 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Output display with markdown formatting
     output = gr.Markdown(label="Tutor's Response")
 
-    # Trigger action
+    # Loading indicator
+    with gr.Row():
+        loading = gr.Markdown("Loading...", visible=False)
+
+    # Submit button with loading feedback
     submit_button = gr.Button("Get Started", variant="primary")
     submit_button.click(
-        tutor_interface,
+        fn=tutor_interface,
         inputs=[name, experience, goals, topic, request_challenge],
-        outputs=output
+        outputs=[output],
+        _js="""
+        () => {
+            document.querySelector('[id^=loading]').style.display = 'block';
+            return null;
+        }
+        """
+    ).then(
+        fn=lambda: gr.update(visible=False),
+        inputs=None,
+        outputs=loading
     )
 
 # Launch the app