Update app.py

app.py

@@ -457,7 +457,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
 demo.launch()'''
 
-import gradio as gr
+'''import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import sys
@@ -577,6 +577,33 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     exit_check.change(fn=lambda x: sys.exit() if x else None, inputs=[exit_check])
 
-demo.launch()
+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}"}
+
+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}"
+    
+    payload = {"inputs": prompt}
+    response = requests.post(API_URL, headers=HEADERS, json=payload)
+    
+    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", "", ""
+