Update app.py

app.py

@@ -10,7 +10,7 @@ def smart_response(command, execute):
     # Dynamically adjust max_new_tokens based on complexity
     simple_keywords = ["open", "close", "shutdown", "restart", "lock", "screenshot"]
     is_simple = any(keyword in command.lower() for keyword in simple_keywords)
-    max_tokens = 128 if is_simple else 1024  # fast for simple, detailed for complex
+    max_tokens = 128 if is_simple else 1024  # Shorter for quick commands, longer for reasoning
 
     return get_assistant_response(pipe, command, execute, max_new_tokens=max_tokens)
 
@@ -18,16 +18,25 @@ def smart_response(command, execute):
 demo = gr.Interface(
     fn=smart_response,
     inputs=[
-        gr.Textbox(lines=2, placeholder="e.g. Open Chrome or Take a screenshot", label="🧾 Command"),
+        gr.Textbox(
+            lines=2,
+            placeholder="e.g. Open Chrome or Take a screenshot",
+            label="🧾 Command"
+        ),
         gr.Checkbox(label="🛠️ Execute command (if possible)")
     ],
     outputs="text",
     title="🧠 Smart AI Assistant",
     description=(
-        "AI assistant powered by google/medgemma-4b-it\n\n"
-        "Try commands like: 'Open Chrome', 'Take a screenshot', 'Create a file', or 'What is AI?'\n"
-        "✔️ Short commands = fast responses | 🧠 Big questions = detailed answers\n"
-        "✅ Enable the checkbox to let it run real OS commands (where allowed)"
+        "AI assistant powered by **Mistral-7B-Instruct** 🧠\n\n"
+        "Try commands like:\n"
+        "• 'Open Chrome'\n"
+        "• 'Take a screenshot'\n"
+        "• 'What is AGI?'\n"
+        "• 'Summarize the history of AI'\n\n"
+        "✅ Simple commands → fast replies\n"
+        "🧠 Complex questions → thoughtful answers\n"
+        "⚠️ Check the box to allow real OS command execution"
     ),
     flagging_mode="never"
 )