Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Custom background CSS with semi-transparent panel
+# Background CSS
 css = """
 body {
   background-image: url('https://cdn-uploads.huggingface.co/production/uploads/67351c643fe51cb1aa28f2e5/YcsJnPk8HJvXiB5WkVmf1.jpeg');
@@ -28,99 +28,84 @@ body {
   box-shadow: 0 0 12px rgba(0, 0, 0, 0.3);
   overflow-y: auto;
 }
-/* Improved title styling */
 .gradio-container .chatbot h1 {
    color: var(--custom-title-color) !important;
    font-family: 'Noto Sans', serif !important;
-   font-size: 5rem !important;  /* Increased font size */
+   font-size: 5rem !important;
    font-weight: bold !important;
    text-align: center !important;
    margin-bottom: 1.5rem !important;
-   width: 100%;  /* Ensure full width for centering */
+   width: 100%;
 }
 """
 
+# Model client (consider switching to a public model like mistralai if 401 persists)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def respond(
-    message,
-    history,
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
+# Level prompt selector
+def level_to_prompt(level):
+    return {
+        "A1": "You are a friendly French tutor. Speak mostly in English, use simple French, and explain everything.",
+        "A2": "You are a patient French tutor. Use short French phrases and explain them in English.",
+        "B1": "You are a helpful French tutor. Speak mostly in French but clarify in English when needed.",
+        "B2": "You are a French tutor. Speak primarily in French with rare English support.",
+        "C1": "You are a native French tutor. Speak entirely in French, clearly and professionally.",
+        "C2": "You are a native French professor. Speak in rich, complex French. Avoid English."
+    }.get(level, "You are a helpful French tutor.")
+
+# Chat handler
+def respond(message, history, user_level, max_tokens, temperature, top_p):
+    system_message = level_to_prompt(user_level)
     messages = [{"role": "system", "content": system_message}]
     
-    # Handle history based on its format (tuple format or messages format)
+    # Handle history
     if history and isinstance(history[0], tuple):
-        # Old tuple format
         for user_msg, assistant_msg in history:
             if user_msg:
                 messages.append({"role": "user", "content": user_msg})
             if assistant_msg:
                 messages.append({"role": "assistant", "content": assistant_msg})
     else:
-        # New messages format
         messages.extend(history)
-    
+
     messages.append({"role": "user", "content": message})
     
     response = ""
-    
-    for msg in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = msg.choices[0].delta.content
-        response += token
-        yield response
+    try:
+        for msg in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            token = msg.choices[0].delta.content
+            if token:
+                response += token
+            yield response
+    except Exception as e:
+        yield f"Désolé! There was an error: {str(e)}"
 
+# Gradio interface
 with gr.Blocks(css=css) as demo:
-    # Title Markdown block
     gr.Markdown("French Instructor", elem_id="custom-title")
     
     with gr.Column(elem_id="chat-panel"):
-        with gr.Accordion("Advanced Settings", open=False):
-            system_message = gr.Textbox(
-                value="You are a helpful French language tutor. You help users learn French vocabulary, grammar, and cultural contexts. When appropriate, include both the French writing and pronunciation. For beginners, focus on simple phrases and gradually increase difficulty based on user proficiency.", 
-                label="System Message"
+        with gr.Accordion("⚙️ Advanced Settings", open=False):
+            user_level = gr.Dropdown(
+                choices=["A1", "A2", "B1", "B2", "C1", "C2"],
+                value="A1",
+                label="Your French Level (CEFR)"
             )
-            max_tokens = gr.Slider(
-                minimum=1, 
-                maximum=2048, 
-                value=512, 
-                step=1, 
-                label="Response Length"
-            )
-            temperature = gr.Slider(
-                minimum=0.1, 
-                maximum=4.0, 
-                value=0.7, 
-                step=0.1, 
-                label="Creativity"
-            )
-            top_p = gr.Slider(
-                minimum=0.1, 
-                maximum=1.0, 
-                value=0.95, 
-                step=0.05, 
-                label="Dynamic Text"
-            )
-        
+            max_tokens = gr.Slider(1, 2048, value=512, step=1, label="Response Length")
+            temperature = gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Creativity")
+            top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Dynamic Text Sampling")
+
         gr.ChatInterface(
             respond,
-            additional_inputs=[
-                system_message, 
-                max_tokens, 
-                temperature, 
-                top_p
-            ],
-            type="messages"  # Set to new message format
+            additional_inputs=[user_level, max_tokens, temperature, top_p],
+            type="messages"
         )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()