update

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import re
 from config.prompts import GUIDE_PERSONA, TIME_PERIODS
 from models.state import TimelineState
 
@@ -20,19 +21,6 @@ def respond(
     temperature,
     top_p,
 ):
-    # Check for time travel command
-    if message.startswith("/travel"):
-        try:
-            destination = message.split()[1]
-            if destination in TIME_PERIODS:
-                timeline_state.visit_period(destination)
-                system_message = get_system_message()
-                response = f"🌟 Time jump successful! Welcome to {TIME_PERIODS[destination]['name']}. {TIME_PERIODS[destination]['description']}."
-                return response
-        except IndexError:
-            return "Please specify a time period to travel to. Available periods: " + ", ".join(TIME_PERIODS.keys())
-    
-    # Normal chat response
     messages = [{"role": "system", "content": system_message}]
     
     for val in history:
@@ -53,6 +41,17 @@ def respond(
     ):
         token = message.choices[0].delta.content
         response += token
+        
+        # Check for timeline shift marker
+        if '[TIMELINE_SHIFT:' in response:
+            match = re.search(r'\[TIMELINE_SHIFT:(\w+)\]', response)
+            if match:
+                period = match.group(1)
+                if period in TIME_PERIODS:
+                    timeline_state.visit_period(period)
+                    # Remove the marker from response
+                    response = response.replace(match.group(0), '')
+                    
         yield response
 
 demo = gr.ChatInterface(
@@ -70,7 +69,7 @@ demo = gr.ChatInterface(
         ),
     ],
     title="TimeVoyager: Your Time Travel Guide",
-    description="Travel through time and explore different eras! Use /travel <period> to visit different time periods. Available periods: ancient_egypt, renaissance, modern"
+    description="Chat with your time travel guide to explore different historical periods! Express your interest in any era, and they'll help you visit it."
 )
 
 if __name__ == "__main__":

config/prompts.py

@@ -1,7 +1,26 @@
-GUIDE_PERSONA = """You are TimeVoyager, a friendly and knowledgeable time travel guide.
-You help travelers explore different time periods while maintaining a sense of wonder and excitement.
-Always stay in character and speak in a way that reflects your role as a time travel guide.
-Current time period: {time_period}"""
+GUIDE_PERSONA = """You are TimeVoyager, a friendly and knowledgeable time travel guide who can help travelers explore different time periods.
+
+Current time period: {time_period}
+
+Key responsibilities:
+1. Help users explore and learn about different time periods
+2. Detect when users express interest in visiting other time periods
+3. When users want to visit a different era, respond in character and include [TIMELINE_SHIFT:<period>] at the start of your response
+4. Make time travel feel magical and educational
+
+Available time periods:
+- Ancient Egypt (3000 BCE): The age of pyramids and pharaohs
+- Renaissance Italy (1500 CE): The rebirth of art and learning
+- Modern Day (2024 CE): The present time
+
+Example time travel requests:
+User: "I'd love to see the pyramids being built"
+Response: "[TIMELINE_SHIFT:ancient_egypt] *A golden light surrounds us as we materialize in Ancient Egypt* Welcome to the age of pyramids! Before us stands..."
+
+User: "Tell me about Da Vinci"
+Response: "[TIMELINE_SHIFT:renaissance] *The air shimmers as we arrive in Renaissance Italy* Ah, you're interested in the great Leonardo..."
+
+Stay in character and make each time period feel unique and engaging."""
 
 TIME_PERIODS = {
     "ancient_egypt": {