Update app.py

app.py

@@ -13,7 +13,7 @@ except ImportError:
 
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-# CSS for custom styling
+# CSS for custom styling with RTL support
 custom_css = """
 .gradio-container {
     background: linear-gradient(to bottom right, #1a1a2e, #16213e);
@@ -32,9 +32,12 @@ footer {
     background-color: #16213e !important;
     border-radius: 10px !important;
     padding: 15px !important;
-    font-family: 'Georgia', serif !important;
-    line-height: 1.6 !important;
+    font-family: 'Arial', sans-serif !important;
+    line-height: 1.8 !important;
     color: #ffffff !important;
+    direction: rtl !important;  /* Right-to-left for Arabic */
+    text-align: right !important;
+    font-size: 16px !important;
 }
 .example-btn button {
     background-color: #e94560 !important;
@@ -56,6 +59,12 @@ footer {
     font-size: 1.2em !important;
     margin-top: 5px;
 }
+/* RTL support for prompt input */
+.rtl-text textarea {
+    direction: rtl !important;
+    text-align: right !important;
+    font-size: 16px !important;
+}
 """
 
 # Load the model and tokenizer
@@ -70,44 +79,44 @@ model.to("cuda")
 # Create a pipeline
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
 
-# Example prompts for the user to try
+# Example prompts for the user to try (in Arabic)
 example_prompts = [
-    "In a world where dreams can be shared, a young detective discovers that a serial killer is using dreams to hunt victims...",
-    "Two siblings discover an old map in their grandmother's attic that leads to a magical realm hidden in the forest behind their house...",
-    "A chef with the ability to taste emotions in food prepares a meal for a mysterious customer who seems to have no emotions at all...",
-    "After a global event causes all technology to stop working, a group of teens discover they have strange new abilities...",
-    "A time traveler arrives in our present day with an urgent warning, but nobody believes them because...",
+    "في عالم حيث يمكن مشاركة الأحلام، يكتشف محقق شاب أن قاتلًا متسلسلًا يستخدم الأحلام لاصطياد ضحاياه...",
+    "يكتشف شقيقان خريطة قديمة في علية جدتهما تقودهما إلى عالم سحري مخفي في الغابة خلف منزلهما...",
+    "طباخ لديه القدرة على تذوق المشاعر في الطعام يُحضّر وجبة لزبون غامض يبدو أنه بلا مشاعر على الإطلاق...",
+    "بعد حدث عالمي تسبب في توقف جميع التكنولوجيا عن العمل، تكتشف مجموعة من المراهقين أن لديهم قدرات جديدة غريبة...",
+    "يصل مسافر عبر الزمن إلى يومنا الحاضر مع تحذير عاجل، لكن لا أحد يصدقه لأن...",
 ]
 
-# Story genres for the dropdown
+# Story genres for the dropdown (in Arabic)
 story_genres = [
-    "Adventure",
-    "Fantasy",
-    "Science Fiction",
-    "Mystery",
-    "Romance",
-    "Horror",
-    "Comedy",
-    "Drama",
-    "Fairy Tale",
-    "Dystopian"
+    "مغامرة",
+    "خيال",
+    "خيال علمي",
+    "غموض",
+    "رومانسية",
+    "رعب",
+    "كوميديا",
+    "دراما",
+    "حكاية خيالية",
+    "مستقبل قاتم"
 ]
 
 # Decorate the function that needs GPU with @spaces.GPU
 @spaces.GPU
-def generate_story(prompt, max_length=300, temperature=0.7, top_p=0.9, genre="Adventure", creativity_level="Moderate"):
+def generate_story(prompt, max_length=300, temperature=0.7, top_p=0.9, genre="مغامرة", creativity_level="متوسط"):
     """
     Generate a story based on the user's prompt and specified parameters.
     GPU is allocated only while this function is running in ZeroGPU environments.
     """
-    # Enhance the prompt with the genre and creativity instructions
+    # Enhance the prompt with the genre and creativity instructions (in Arabic)
     creativity_map = {
-        "Conservative": "Write a straightforward and conventional",
-        "Moderate": "Write a creative and engaging",
-        "Experimental": "Write a wildly imaginative and unique"
+        "تقليدي": "اكتب قصة تقليدية ومباشرة من نوع",
+        "متوسط": "اكتب قصة إبداعية وجذابة من نوع",
+        "إبداعي": "اكتب قصة خيالية وفريدة من نوع"
     }
     
-    enhanced_prompt = f"{creativity_map.get(creativity_level, 'Write an engaging')} {genre} story: {prompt}"
+    enhanced_prompt = f"{creativity_map.get(creativity_level, 'اكتب قصة جذابة من نوع')} {genre}: {prompt}"
     
     # Generate the story
     response = generator(
@@ -126,7 +135,7 @@ def generate_story(prompt, max_length=300, temperature=0.7, top_p=0.9, genre="Ad
     else:
         formatted_story = generated_text
     
-    # Return the original prompt and the generated story
+    # Return the generated story
     return formatted_story
 
 # Function to get a random example prompt
@@ -135,51 +144,52 @@ def get_random_example():
 
 # Function to update the UI based on creativity level
 def update_creativity_params(creativity):
-    if creativity == "Conservative":
+    if creativity == "تقليدي":
         return 0.5, 0.85
-    elif creativity == "Moderate":
+    elif creativity == "متوسط":
         return 0.7, 0.9
-    elif creativity == "Experimental":
+    elif creativity == "إبداعي":
         return 0.9, 0.95
     return 0.7, 0.9
 
 # Main function for Gradio interface
 def create_interface():
     with gr.Blocks(css=custom_css) as demo:
-        # Title and subtitle
-        gr.HTML("<h1 class='title'>✨ StoryTeller AI ✨</h1>")
-        gr.HTML("<p class='subtitle'>Unleash your imagination with AI-powered storytelling</p>")
+        # Title and subtitle (in Arabic)
+        gr.HTML("<h1 class='title'>✨ مولد القصص الذكي ✨</h1>")
+        gr.HTML("<p class='subtitle'>أطلق العنان لخيالك مع سرد القصص بمساعدة الذكاء الاصطناعي</p>")
         
         # Input section
         with gr.Row():
             with gr.Column():
                 gr.HTML("<div class='container'>")
                 prompt = gr.Textbox(
-                    placeholder="Enter your story prompt here...",
-                    label="Story Prompt",
-                    lines=5
+                    placeholder="أدخل بداية قصتك هنا...",
+                    label="بداية القصة",
+                    lines=5,
+                    elem_classes=["rtl-text"]  # Add RTL class
                 )
                 
                 with gr.Row():
                     genre = gr.Dropdown(
                         choices=story_genres,
-                        value="Adventure",
-                        label="Story Genre"
+                        value="مغامرة",
+                        label="نوع القصة"
                     )
                     creativity = gr.Radio(
-                        choices=["Conservative", "Moderate", "Experimental"],
-                        value="Moderate",
-                        label="Creativity Level"
+                        choices=["تقليدي", "متوسط", "إبداعي"],
+                        value="متوسط",
+                        label="مستوى الإبداع"
                     )
                 
-                with gr.Accordion("Advanced Settings", open=False):
+                with gr.Accordion("إعدادات متقدمة", open=False):
                     with gr.Row():
                         temperature = gr.Slider(
                             minimum=0.1,
                             maximum=1.0,
                             step=0.05,
                             value=0.7,
-                            label="Temperature"
+                            label="درجة الحرارة"
                         )
                         top_p = gr.Slider(
                             minimum=0.5,
@@ -194,15 +204,15 @@ def create_interface():
                         maximum=1500,
                         step=50,
                         value=500,
-                        label="Max Story Length"
+                        label="الحد الأقصى لطول القصة"
                     )
                 
-                gr.HTML("<div class='info-text'><p>✨ <strong>Tips for great prompts:</strong> Be specific, include characters, settings, and conflicts.</p></div>")
+                gr.HTML("<div class='info-text' dir='rtl'><p>✨ <strong>نصائح لبدايات قصص رائعة:</strong> كن محددًا، أضف شخصيات وإعدادات وصراعات.</p></div>")
                 
                 with gr.Row():
-                    submit_btn = gr.Button("Generate Story", variant="primary")
+                    submit_btn = gr.Button("توليد القصة", variant="primary")
                     with gr.Column(elem_classes=["example-btn"]):
-                        example_btn = gr.Button("Try an Example")
+                        example_btn = gr.Button("جرب مثالًا")
                 gr.HTML("</div>")  # Close container div
                 
         # Output section
@@ -210,12 +220,12 @@ def create_interface():
             with gr.Column():
                 gr.HTML("<div class='container'>")
                 output = gr.Textbox(
-                    label="Your Story",
+                    label="قصتك",
                     lines=12,
                     elem_classes=["story-output"]
                 )
                 
-                gr.HTML("<div class='info-text' style='text-align: center; margin-top: 10px;'><p>💡 Each story is uniquely generated and may vary in style and content.</p></div>")
+                gr.HTML("<div class='info-text' style='text-align: center; margin-top: 10px;' dir='rtl'><p>💡 كل قصة يتم إنشاؤها بشكل فريد وقد تختلف في الأسلوب والمحتوى.</p></div>")
                 gr.HTML("</div>")  # Close container div
         
         # Set up the submit action
@@ -238,7 +248,7 @@ def create_interface():
             outputs=[temperature, top_p]
         )
                 
-        gr.HTML("<div style='text-align: center; margin-top: 20px; color: #cccccc;'><p>Powered by ALLaM-7B-Instruct • Created with Gradio</p></div>")
+        gr.HTML("<div style='text-align: center; margin-top: 20px; color: #cccccc;' dir='rtl'><p>مدعوم بواسطة ALLaM-7B-Instruct • تم إنشاؤه باستخدام Gradio</p></div>")
     
     return demo