Create app.py

app.py

@@ -0,0 +1,44 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+# Using a powerful model for creative text generation
+MODEL = "google/flan-t5-large"
+tokenizer = AutoTokenizer.from_pretrained(MODEL)
+model = AutoModelForSeq2SeqLM.from_pretrained(MODEL)
+
+def generate_story(topic, style_choice, lang_choice):
+    # Language and style selection based on user input
+    if lang_choice == "Arabic":
+        if style_choice == "Blog Post (Descriptive)":
+            prompt = f"اكتب مقالاً مفصلاً لمدونة باللغة العربية عن: {topic}. اجعل النغمة شخصية، ووصف الأماكن والأحداث بشكل جذاب."
+        elif style_choice == "Social Media Post (Short & Catchy)":
+            prompt = f"اكتب منشوراً قصيراً وجذاباً باللغة العربية عن: {topic}، مع إيموجي وهاشتاغات مقترحة."
+        else: # Video Script (Storytelling)
+            prompt = f"اكتب سيناريو فيديو (script) باللغة العربية عن: {topic}. ركز على سرد قصة عاطفية وجذابة، مع وصف للمشاهد."
+    else: # English
+        if style_choice == "Blog Post (Descriptive)":
+            prompt = f"Write a detailed blog post in English about: {topic}. Make the tone personal, and describe the places and events attractively."
+        elif style_choice == "Social Media Post (Short & Catchy)":
+            prompt = f"Write a short and catchy social media post in English about: {topic}, with emojis and suggested hashtags."
+        else: # Video Script (Storytelling)
+            prompt = f"Write a video script (storytelling) in English about: {topic}. Focus on a personal and emotional story, with scene descriptions."
+
+    try:
+        inputs = tokenizer(prompt, return_tensors="pt")
+        outs = model.generate(**inputs, max_length=500, num_beams=5, early_stopping=True)
+        return tokenizer.decode(outs[0], skip_special_tokens=True)
+    except Exception as e:
+        return f"An error occurred: {str(e)}. Please make sure the topic is clearly written."
+
+iface = gr.Interface(
+    fn=generate_story,
+    inputs=[
+        gr.Textbox(label="Topic (e.g., 'My trip to Paris, seeing the Eiffel Tower and eating croissants')"),
+        gr.Radio(["Blog Post (Descriptive)", "Social Media Post (Short & Catchy)", "Video Script (Storytelling)"], label="Style"),
+        gr.Radio(["Arabic", "English"], label="Language")
+    ],
+    outputs=gr.Textbox(label="Result", lines=10)
+)
+
+if __name__ == "__main__":
+    iface.launch()