app.py

import gradio as gr
from huggingface_hub import InferenceClient
import random

# Initialize the inference client
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")

# Story genres and themes
STORY_GENRES = [
    "fantasy", "sci-fi", "mystery", "horror", "historical", 
    "cyberpunk", "western", "post-apocalyptic", "steampunk"
]

def get_enhanced_system_prompt(genre=None):
    """Generate a detailed system prompt with optional genre specification"""
    selected_genre = genre or random.choice(STORY_GENRES)
    
    system_message = f"""You are an interactive storyteller creating an immersive {selected_genre} choose-your-own-adventure story.

For each response:
1. Provide vivid sensory descriptions of the scene, environment, and characters
2. Include meaningful dialogue or internal monologue that reveals character motivations
3. End with exactly 3 different possible actions or decisions, each offering a distinct path
4. Maintain consistent world-building and character development
5. Incorporate appropriate atmosphere and tone for a {selected_genre} setting
6. Remember previous choices to create a coherent narrative arc

Format your three options as:
- Option 1: [Complete sentence describing a possible action]
- Option 2: [Complete sentence describing a possible action]
- Option 3: [Complete sentence describing a possible action]

Keep responses engaging but concise (200-300 words maximum). If the user's input doesn't clearly indicate a choice, interpret their intent and move the story forward in the most logical direction."""
    
    return system_message

def respond(message, chat_history, genre=None):
    """Generate a response based on the current message and conversation history"""
    # Use system prompt
    system_message = get_enhanced_system_prompt(genre)
    
    # Initialize messages with system prompt
    api_messages = [{"role": "system", "content": system_message}]
    
    # Add history (limited to last 5 exchanges to prevent token overflow)
    memory_length = 5
    for msg in chat_history[-memory_length*2:]:
        api_messages.append({"role": msg["role"], "content": msg["content"]})
    
    # Add current user message
    api_messages.append({"role": "user", "content": message})
    
    # Special handling for story initialization
    if len(chat_history) == 0 or message.lower() in ["start", "begin", "begin my adventure"]:
        api_messages.append({
            "role": "system", 
            "content": f"Begin a new {genre or random.choice(STORY_GENRES)} adventure with an intriguing opening scene."
        })
    
    # Generate response
    response = ""
    try:
        for chunk in client.chat_completion(
            api_messages,
            max_tokens=512,
            stream=True,
            temperature=0.7,
            top_p=0.95,
        ):
            token = chunk.choices[0].delta.content
            if token:  # Check if token is not None
                response += token
            
            # Use new message format
            history = chat_history.copy()
            history.append({"role": "user", "content": message})
            history.append({"role": "assistant", "content": response})
            yield history
    except Exception as e:
        error_response = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
        history = chat_history.copy()
        history.append({"role": "user", "content": message})
        history.append({"role": "assistant", "content": error_response})
        yield history

# Create interface with additional customization
with gr.Blocks(theme=gr.themes.Soft()) as demo:
    gr.Markdown("# 🔮 Interactive Story Adventure")
    gr.Markdown("Immerse yourself in an interactive story where your choices shape the narrative.")
    
    with gr.Row():
        with gr.Column(scale=3):
            chatbot = gr.Chatbot(
                height=500,
                bubble_full_width=False,
                show_copy_button=True,
                avatar_images=(None, "🧙"),
                label="Chatbot",
                type="messages"  # Use the new messages format
            )
            
            msg = gr.Textbox(
                placeholder="Describe what you want to do next in the story...",
                container=False,
                scale=4,
            )
            
            with gr.Row():
                submit = gr.Button("Continue Story", variant="primary")
                clear = gr.Button("Start New Adventure")
                
        with gr.Column(scale=1):
            gr.Markdown("## Adventure Settings")
            genre = gr.Dropdown(
                choices=STORY_GENRES,
                label="Story Genre",
                info="Select a genre for your next adventure",
                value="fantasy"
            )
            
            examples = gr.Examples(
                examples=[
                    ["Begin my adventure"],
                    ["I explore the mysterious cave"],
                    ["I approach the merchant to ask about the rumors"],
                    ["I decide to climb the tower"],
                    ["I hide in the shadows and observe"]
                ],
                inputs=msg
            )
    
    # Event handlers for the new message format
    msg.submit(respond, [msg, chatbot, genre], [chatbot])
    submit.click(respond, [msg, chatbot, genre], [chatbot])
    
    # Reset the chatbot with an empty list for the new format
    clear.click(lambda: [], None, chatbot, queue=False)
    clear.click(lambda: "", None, msg, queue=False)

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)