Update app.py

app.py

@@ -6,41 +6,48 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 MODEL_ID = "ApolloAIs/phoebus-3.1-1.5b-merged"
 
-print("Loading Phoebus 3.1...")
+print("Initializing Phoebus 3.1 Flash engine...", flush=True)
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
-    dtype=torch.float16,
+    torch_dtype=torch.float16,
     device_map="auto"
 )
-print("Model ready!")
+print("Phoebus 3.1 Flash ready!", flush=True)
 
 def format_emem_cell64(claim: str) -> str:
     encoded = base64.b64encode(claim.encode("utf-8")).decode("utf-8")[:12]
     return f"emem:fact:cell64_{encoded}"
 
-@spaces.GPU
+@spaces.GPU(duration=30)
 def phoebus_respond(message, history, enable_tinyfish, enable_emem):
     system_prompt = (
-        "You are Phoebus 3.1, a reasoning model specialized in deep space telemetry, "
+        "You are Phoebus 3.1 Flash, a rapid-reasoning model specialized in deep space telemetry, "
         "Earth geography, and coding, developed by Clovr (atclovr.dev). "
         "Always cite emem protocol receipts when validating spatial coordinates."
     )
     
     context = ""
-    if enable_tinyfish or "search" in message.lower():
-        context += f"\n\n[System Context - TinyFish Data]: Verified telemetry and spatial data confirmed for '{message}'."
+    msg_text = message if isinstance(message, str) else message.get("text", "")
+    
+    if enable_tinyfish or "search" in msg_text.lower():
+        context += f"\n\n[System Context - TinyFish Data]: Verified telemetry and spatial data confirmed for '{msg_text}'."
         
-    if enable_emem or any(w in message.lower() for w in ["location", "coordinates", "where"]):
-        receipt = format_emem_cell64(message)
+    if enable_emem or any(w in msg_text.lower() for w in ["location", "coordinates", "where"]):
+        receipt = format_emem_cell64(msg_text)
         context += f"\n\n[System Protocol - Vortx AI emem]: Attach receipt `{receipt}` for physical spatial validation."
 
+    # Build conversation messages supporting both dict and tuple history formats
     formatted_messages = [{"role": "system", "content": system_prompt}]
-    for user_turn, bot_turn in history:
-        formatted_messages.append({"role": "user", "content": user_turn})
-        formatted_messages.append({"role": "assistant", "content": bot_turn})
+    
+    for turn in history:
+        if isinstance(turn, dict):
+            formatted_messages.append({"role": turn.get("role", "user"), "content": str(turn.get("content", ""))})
+        elif isinstance(turn, (list, tuple)) and len(turn) == 2:
+            formatted_messages.append({"role": "user", "content": str(turn[0])})
+            formatted_messages.append({"role": "assistant", "content": str(turn[1])})
         
-    formatted_messages.append({"role": "user", "content": message + context})
+    formatted_messages.append({"role": "user", "content": msg_text + context})
 
     prompt = tokenizer.apply_chat_template(
         formatted_messages,
@@ -62,7 +69,7 @@ def phoebus_respond(message, history, enable_tinyfish, enable_emem):
     return reply
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
-    gr.Markdown("# 🛰️ Phoebus 3.1 Console\n*Reasoning model by Clovr (atclovr.dev)*")
+    gr.Markdown("# ⚡ Phoebus 3.1 Flash Console\n*Ultra-fast reasoning model by Clovr (atclovr.dev)*")
     
     with gr.Row():
         tinyfish_toggle = gr.Checkbox(label="Enable TinyFish Search", value=True)
@@ -72,7 +79,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
         fn=phoebus_respond,
         additional_inputs=[tinyfish_toggle, emem_toggle],
         chatbot=gr.Chatbot(height=500),
-        textbox=gr.Textbox(placeholder="Ask Phoebus 3.1 about telemetry, geography, or code...", container=False, scale=7),
+        textbox=gr.Textbox(placeholder="Ask Phoebus 3.1 Flash anything...", container=False, scale=7),
     )
 
 if __name__ == "__main__":