Create AI Based Story Teller

AI Based Story Teller

@@ -0,0 +1,34 @@
+pip install transformers gradio
+import gradio as gr
+from transformers import pipeline
+
+# Load the GPT model for text generation
+generator = pipeline('text-generation', model='gpt-3.5-turbo')
+
+# Define the function to generate a story
+def generate_story(genre, prompt):
+    if genre == "Fantasy":
+        prompt = f"A fantasy story about {prompt}."
+    elif genre == "Sci-Fi":
+        prompt = f"A science fiction story about {prompt}."
+    elif genre == "Romance":
+        prompt = f"A romantic story about {prompt}."
+    else:
+        prompt = f"A story about {prompt}."
+    
+    story = generator(prompt, max_length=500, num_return_sequences=1)[0]['generated_text']
+    return story
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=generate_story,
+    inputs=[
+        gr.Dropdown(choices=["Fantasy", "Sci-Fi", "Romance", "Mystery"], label="Select Genre"),
+        gr.Textbox(label="Enter a prompt", placeholder="Enter a theme, character name, etc."),
+    ],
+    outputs=gr.Textbox(label="Generated Story"),
+    live=True
+)
+
+# Launch the interface
+iface.launch()