Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 from google import genai
 import gradio as gr
 
@@ -7,14 +6,16 @@ API_KEY = "AIzaSyB4JKubDJd7nLx1NqPhDfMGeVWeQ7kqClY"
 client = genai.Client(api_key=API_KEY)
 MODEL_NAME = "gemini-2.5-flash"
 
-
-def generate_main_question_gemini(paragraph: str):
-    if not paragraph or paragraph.strip() == "":
-        return "الرجاء إدخال فقرة أولاً."
-
+def generate_study_schedule(subjects, lessons, days, hours_per_day):
+    # التحقق من الإدخالات
+    if not subjects.strip() or not lessons.strip() or not days.strip() or not hours_per_day.strip():
+        return "الرجاء ملء جميع الحقول أولاً."
+    
     prompt = f"""
-    بناءا على 
-{paragraph}
+المواد الدراسية: {subjects}
+عدد الدروس أو الفصول لكل مادة: {lessons}
+عدد الأيام المتاحة للدراسة: {days}
+عدد الساعات المتاحة للدراسة في كل يوم: {hours_per_day}
 
 المطلوب:
 
@@ -31,23 +32,44 @@ def generate_main_question_gemini(paragraph: str):
         response = client.models.generate_content(model=MODEL_NAME, contents=prompt)
         return response.text.strip()
     except Exception as e:
-        return f" Error while connecting to API: {e}"
+        return f"Error while connecting to API: {e}"
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("## MainQuestion — Basic Question Generator (Arabic Output)")
-    
+    gr.Markdown("## جدول الدراسة — منظم تلقائي (Arabic Output)")
+
     with gr.Row():
-        paragraph = gr.Textbox(
-            label="Paragraph (Input text)", 
-            lines=8, 
-            placeholder="Paste the paragraph here..."
+        subjects_input = gr.Textbox(
+            label="أسماء المواد الدراسية",
+            placeholder="مثال: رياضيات، فيزياء، كيمياء",
+            lines=2
         )
-    
-    output = gr.Textbox(label="Generated Question (Arabic)", lines=3)
+        lessons_input = gr.Textbox(
+            label="عدد الدروس أو الفصول في كل مادة",
+            placeholder="مثال: 10، 8، 12",
+            lines=2
+        )
+
+    with gr.Row():
+        days_input = gr.Textbox(
+            label="عدد الأيام المتاحة للدراسة",
+            placeholder="مثال: 7",
+            lines=1
+        )
+        hours_input = gr.Textbox(
+            label="عدد الساعات المتاحة للدراسة في كل يوم",
+            placeholder="مثال: 4",
+            lines=1
+        )
+
+    output = gr.Textbox(label="الجدول الناتج (باراغراف)", lines=10)
 
-    submit_btn = gr.Button("Submit")
-    submit_btn.click(fn=generate_main_question_gemini, inputs=paragraph, outputs=output)
+    submit_btn = gr.Button("إنشاء الجدول")
+    submit_btn.click(
+        fn=generate_study_schedule, 
+        inputs=[subjects_input, lessons_input, days_input, hours_input], 
+        outputs=output
+    )
 
 if __name__ == "__main__":
-    demo.launch(share=True, show_error=True)
+    demo.launch(share=True, show_error=True)