Update app.py

app.py

@@ -579,31 +579,124 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
 demo.launch()'''
 
-import requests
-
-API_TOKEN = "Sakshi"
-API_URL = f"https://api-inference.huggingface.co/models/TinyLlama/TinyLlama-1.1B-Chat-v1.0"
-HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
+import gradio as gr
+from transformers import pipeline
 
-def generate_story_continuation(current_story, selected_option=None):
-    """
-    Generate story continuation using Hugging Face API
-    """
-    if selected_option:
-        prompt = f"""Previous story: {current_story}
-        Selected path: {selected_option}
-        Continue the story with this choice and provide two new options."""
-    else:
-        prompt = f"Create a story based on: {current_story}"
+# Load a lightweight text generation model from Hugging Face
+generator = pipeline("text-generation", model="distilgpt2", max_length=100)
+
+# Initialize story state
+story_history = []
+
+def generate_story_segment(prompt, user_choice=None):
+    """Generate a story segment and three choices based on the prompt."""
+    if user_choice:
+        prompt = f"{prompt} {user_choice}"
+    
+    # Generate a story segment
+    story_output = generator(prompt, max_length=100, num_return_sequences=1, do_sample=True, temperature=0.7)[0]["generated_text"]
+    
+    # Generate three possible choices for the next step
+    choice_prompt = f"{story_output} What happens next? Provide three short options."
+    choices_output = generator(choice_prompt, max_length=150, num_return_sequences=3, do_sample=True, temperature=0.9)
+    
+    # Extract and clean choices
+    choices = [choice["generated_text"].split("Provide three short options.")[-1].strip() for choice in choices_output]
+    cleaned_choices = []
+    for choice in choices:
+        # Simple heuristic to extract short options (could be improved with regex or parsing)
+        if len(choice) > 0:
+            cleaned_choices.append(choice[:50] + "..." if len(choice) > 50 else choice)
+        else:
+            cleaned_choices.append("Something unexpected happens...")
+    
+    # Ensure we always have 3 choices
+    while len(cleaned_choices) < 3:
+        cleaned_choices.append("A new adventure begins...")
+    
+    return story_output, cleaned_choices[:3]
+
+def start_story(initial_prompt):
+    """Start the story with the user's initial prompt."""
+    global story_history
+    story_history = [initial_prompt]
+    segment, choices = generate_story_segment(initial_prompt)
+    story_history.append(segment)
+    return "\n".join(story_history), choices
+
+def continue_story(choice):
+    """Continue the story based on the user's choice."""
+    global story_history
+    segment, choices = generate_story_segment(story_history[-1], choice)
+    story_history.append(f"You chose: {choice}")
+    story_history.append(segment)
+    return "\n".join(story_history), choices
+
+def reset_story():
+    """Reset the story to start fresh."""
+    global story_history
+    story_history = []
+    return "Story reset. Enter a new prompt to begin!", ["", "", ""]
+
+# Gradio interface
+with gr.Blocks(title="Interactive Story Generator") as demo:
+    gr.Markdown("# Interactive Story Generator")
+    gr.Markdown("Start your story by entering a prompt below. Then, choose options to guide the adventure!")
+    
+    # Initial prompt input
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Enter your story prompt (e.g., 'A knight in a dark forest')", placeholder="Type here...")
+        start_button = gr.Button("Start Story")
+    
+    # Story display and choice selection
+    story_output = gr.Textbox(label="Your Story", lines=10, interactive=False)
+    choice_buttons = [
+        gr.Button(value="", visible=True),
+        gr.Button(value="", visible=True),
+        gr.Button(value="", visible=True)
+    ]
+    
+    # Reset button
+    reset_button = gr.Button("Reset Story")
+    
+    # Logic to start the story
+    def update_choices(story, choices):
+        return story, [
+            gr.Button.update(value=choices[0], visible=True),
+            gr.Button.update(value=choices[1], visible=True),
+            gr.Button.update(value=choices[2], visible=True)
+        ]
+    
+    start_button.click(
+        fn=start_story,
+        inputs=prompt_input,
+        outputs=[story_output, choice_buttons[0], choice_buttons[1], choice_buttons[2]],
+        _js="() => {return [document.querySelector('input').value]}"
+    ).then(
+        fn=update_choices,
+        inputs=[story_output, gr.State(value=["", "", ""])],
+        outputs=[story_output, choice_buttons[0], choice_buttons[1], choice_buttons[2]]
+    )
     
-    payload = {"inputs": prompt}
-    response = requests.post(API_URL, headers=HEADERS, json=payload)
+    # Logic to continue the story with choices
+    for i, button in enumerate(choice_buttons):
+        button.click(
+            fn=continue_story,
+            inputs=gr.State(value=button.value),
+            outputs=[story_output, choice_buttons[0], choice_buttons[1], choice_buttons[2]],
+            _js=f"() => ['{button.value}']"
+        ).then(
+            fn=update_choices,
+            inputs=[story_output, gr.State(value=["", "", ""])],
+            outputs=[story_output, choice_buttons[0], choice_buttons[1], choice_buttons[2]]
+        )
     
-    if response.status_code == 200:
-        story = response.json()[0]['generated_text']
-        return story, "Option 1: Continue the adventure", "Option 2: Take a different direction"
-    else:
-        return "Error generating story", "", ""
-
+    # Reset the story
+    reset_button.click(
+        fn=reset_story,
+        outputs=[story_output, choice_buttons[0], choice_buttons[1], choice_buttons[2]]
+    )
 
+# Launch the app
+demo.launch()