Update app.py

app.py

@@ -23,9 +23,10 @@ def format_system_info():
 @retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=4, max=10))
 def initialize_model():
     try:
-        token = os.getenv("HUGGING_FACE_HUB_TOKEN")
+        # Try HUGGING_FACE_HUB_TOKEN first, fallback to HF_TOKEN
+        token = os.getenv("HUGGING_FACE_HUB_TOKEN") or os.getenv("HF_TOKEN")
         if not token:
-            return False, "No token found. Please set HUGGING_FACE_HUB_TOKEN in Space secrets.", None
+            return False, "No token found. Please set HUGGING_FACE_HUB_TOKEN or HF_TOKEN in Space secrets.", None
             
         # Force re-login to refresh connection
         login(token=token, add_to_git_credential=False)
@@ -94,7 +95,7 @@ def chat(message, history):
 
 def check_connection():
     try:
-        token = os.getenv("HUGGING_FACE_HUB_TOKEN")
+        token = os.getenv("HUGGING_FACE_HUB_TOKEN") or os.getenv("HF_TOKEN")
         api = HfApi(token=token)
         model_info = api.model_info("CohereLabs/c4ai-command-a-03-2025")
         return f"""
@@ -108,7 +109,7 @@ Last Modified: {model_info.lastModified}
 
 # Create the Gradio interface with connection monitoring
 with gr.Blocks(theme=gr.themes.Default()) as demo:
-    gr.Markdown(f"# Medical Decision Support AI")
+    gr.Markdown(f"# Medical Decision Support AI\n{format_system_info()}")
     
     with gr.Row():
         connection_btn = gr.Button("Check Connection Status")
@@ -116,13 +117,13 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     
     chat_interface = gr.ChatInterface(
         fn=chat,
-        description="A medical decision support system that provides healthcare-related information and guidance.",
+        description=f"A medical decision support system that provides healthcare-related information and guidance.\n{format_system_info()}",
         examples=[
             "What are the symptoms of hypertension?",
             "What are common drug interactions with aspirin?",
             "What are the warning signs of diabetes?",
         ],
-        type='messages'  # Updated to use new message format
+        type='messages'  # Using new message format
     )
     
     connection_btn.click(check_connection, outputs=connection_status)