Update app.py

app.py

@@ -1,19 +1,19 @@
 import gradio as gr
 import cohere
-import os
 from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
 
 # Initialize Cohere API client
-co = cohere.Client(os.getenv('COHERE_API_KEY'))
+co = cohere.Client()
+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 +22,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 +31,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 +80,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Output display with markdown formatting
     output = gr.Markdown(label="Tutor's Response")
 
-    # Trigger action
+    # Submit button
     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]
     )
 
 # Launch the app