Update app.py

app.py

@@ -7,7 +7,7 @@ from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib.units import inch
 from reportlab.lib import colors
 from reportlab.lib.enums import TA_CENTER, TA_LEFT
-import io
+import tempfile
 import os
 from PIL import Image as PILImage
 
@@ -95,15 +95,17 @@ def generate_schedule(start_date, end_date, off_days, num_lessons, logo_file=Non
 
 def create_pdf(schedule_data, start_date, end_date, off_days, num_lessons, logo_file=None):
     """
-    Create PDF from schedule data
+    Create PDF from schedule data and return file path
     """
     try:
-        # Create a bytes buffer for the PDF
-        buffer = io.BytesIO()
+        # Create a temporary file for the PDF
+        pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf', prefix='study_schedule_')
+        pdf_path = pdf_file.name
+        pdf_file.close()
         
         # Create PDF document
-        doc = SimpleDocTemplate(buffer, pagesize=A4, rightMargin=72, leftMargin=72,
-                              topMargin=72, bottomMargin=18)
+        doc = SimpleDocTemplate(pdf_path, pagesize=A4, rightMargin=72, leftMargin=72,
+                              topMargin=72, bottomMargin=72)
         
         # Container for the 'Flowable' objects
         elements = []
@@ -131,20 +133,21 @@ def create_pdf(schedule_data, start_date, end_date, off_days, num_lessons, logo_
         # Add logo if provided
         if logo_file is not None:
             try:
-                # Save uploaded file temporarily
-                logo_path = "temp_logo.png"
-                with open(logo_path, "wb") as f:
-                    f.write(logo_file)
+                # Create temporary file for logo
+                logo_temp = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+                logo_temp.write(logo_file)
+                logo_temp.close()
                 
                 # Add logo to PDF
-                logo = Image(logo_path, width=2*inch, height=1*inch)
+                logo = Image(logo_temp.name, width=2*inch, height=1*inch)
                 logo.hAlign = 'CENTER'
                 elements.append(logo)
                 elements.append(Spacer(1, 20))
                 
-                # Clean up temp file
-                os.remove(logo_path)
-            except:
+                # Clean up temp logo file
+                os.unlink(logo_temp.name)
+            except Exception as e:
+                print(f"Logo processing error: {e}")
                 pass  # Skip logo if there's an error
         
         # Add title
@@ -194,63 +197,43 @@ def create_pdf(schedule_data, start_date, end_date, off_days, num_lessons, logo_
         
         elements.append(table)
         
+        # Add footer
+        elements.append(Spacer(1, 30))
+        footer = Paragraph(f"Generated on {datetime.now().strftime('%Y-%m-%d %H:%M')}", 
+                          styles['Normal'])
+        elements.append(footer)
+        
         # Build PDF
         doc.build(elements)
         
-        # Get the value of the BytesIO buffer
-        pdf_data = buffer.getvalue()
-        buffer.close()
-        
-        return pdf_data
+        return pdf_path
         
     except Exception as e:
         print(f"PDF creation error: {str(e)}")
         return None
 
-def create_csv(schedule_data):
-    """
-    Create CSV from schedule data
-    """
-    try:
-        df = pd.DataFrame(schedule_data)
-        csv_buffer = io.StringIO()
-        df.to_csv(csv_buffer, index=False)
-        return csv_buffer.getvalue()
-    except Exception as e:
-        print(f"CSV creation error: {str(e)}")
-        return None
-
 def process_schedule(start_date, end_date, off_days, num_lessons, logo_file):
     """
     Main processing function for Gradio interface
     """
-    # Generate schedule
-    summary, df, schedule_data = generate_schedule(start_date, end_date, off_days, num_lessons, logo_file)
-    
-    if schedule_data is None:
-        return summary, None, None, None
-    
-    # Create PDF
-    pdf_data = create_pdf(schedule_data, start_date, end_date, off_days, num_lessons, logo_file)
-    
-    # Create CSV
-    csv_data = create_csv(schedule_data)
-    
-    # Save files temporarily for download
-    pdf_path = None
-    csv_path = None
-    
-    if pdf_data:
-        pdf_path = "study_schedule.pdf"
-        with open(pdf_path, "wb") as f:
-            f.write(pdf_data)
-    
-    if csv_data:
-        csv_path = "study_schedule.csv"
-        with open(csv_path, "w") as f:
-            f.write(csv_data)
-    
-    return summary, df, pdf_path, csv_path
+    try:
+        # Generate schedule
+        summary, df, schedule_data = generate_schedule(start_date, end_date, off_days, num_lessons, logo_file)
+        
+        if schedule_data is None:
+            return summary, None, None, gr.update(visible=False)
+        
+        # Create PDF
+        pdf_path = create_pdf(schedule_data, start_date, end_date, off_days, num_lessons, logo_file)
+        
+        if pdf_path is None:
+            return summary, df, None, gr.update(visible=False)
+        
+        return summary, df, pdf_path, gr.update(visible=True)
+        
+    except Exception as e:
+        error_msg = f"❌ Error processing schedule: {str(e)}"
+        return error_msg, None, None, gr.update(visible=False)
 
 # Create Gradio interface
 def create_interface():
@@ -270,6 +253,15 @@ def create_interface():
             border-left: 4px solid #2E86AB;
             margin: 1rem 0;
         }
+        .generate-btn {
+            background: linear-gradient(45deg, #2E86AB, #A23B72) !important;
+            border: none !important;
+            color: white !important;
+            font-weight: bold !important;
+            font-size: 16px !important;
+            padding: 12px 24px !important;
+            border-radius: 8px !important;
+        }
         """
     ) as demo:
         
@@ -283,7 +275,7 @@ def create_interface():
             2. Choose your off-days (days you don't want to study)<br>
             3. Set the number of lessons to schedule<br>
             4. Optionally upload a logo for your PDF<br>
-            5. Generate your personalized study schedule!
+            5. Generate and download your personalized study schedule as PDF!
             </div>
             """,
             elem_classes=["main-header"]
@@ -294,15 +286,17 @@ def create_interface():
                 gr.Markdown("### 📅 Schedule Settings")
                 
                 start_date = gr.Textbox(
-                    label="Start Date",
+                    label="📅 Start Date",
                     placeholder="YYYY-MM-DD (e.g., 2025-07-01)",
-                    value="2025-07-01"
+                    value="2025-07-01",
+                    info="Enter start date in YYYY-MM-DD format"
                 )
                 
                 end_date = gr.Textbox(
-                    label="End Date", 
+                    label="📅 End Date", 
                     placeholder="YYYY-MM-DD (e.g., 2025-08-30)",
-                    value="2025-08-30"
+                    value="2025-08-30",
+                    info="Enter end date in YYYY-MM-DD format"
                 )
                 
                 num_lessons = gr.Slider(
@@ -310,61 +304,96 @@ def create_interface():
                     maximum=200,
                     value=90,
                     step=1,
-                    label="Number of Lessons"
+                    label="📚 Number of Lessons",
+                    info="Total lessons to schedule"
                 )
                 
                 off_days = gr.CheckboxGroup(
                     choices=["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"],
-                    label="Off Days (Days you don't want to study)",
-                    value=["Sunday"]
+                    label="🚫 Off Days (Days you don't want to study)",
+                    value=["Sunday"],
+                    info="Select days to skip"
                 )
                 
                 logo_file = gr.File(
-                    label="Upload Logo (Optional)",
+                    label="🖼️ Upload Logo (Optional)",
                     file_types=["image"],
-                    type="binary"
+                    type="binary",
+                    info="Upload PNG, JPG, or other image for PDF header"
                 )
                 
-                generate_btn = gr.Button("🚀 Generate Schedule", variant="primary", size="lg")
+                generate_btn = gr.Button(
+                    "🚀 Generate Schedule", 
+                    variant="primary", 
+                    size="lg",
+                    elem_classes=["generate-btn"]
+                )
         
         with gr.Row():
             with gr.Column():
                 gr.Markdown("### 📊 Results")
                 
                 summary_output = gr.Markdown()
+                
                 schedule_table = gr.Dataframe(
                     headers=["Date", "Day", "Session"],
-                    interactive=False
+                    interactive=False,
+                    wrap=True
                 )
                 
-                with gr.Row():
-                    pdf_download = gr.File(label="📄 Download PDF", visible=False)
-                    csv_download = gr.File(label="📊 Download CSV", visible=False)
+                pdf_download = gr.File(
+                    label="📄 Download PDF Schedule", 
+                    visible=False,
+                    interactive=True
+                )
         
         # Event handler
         generate_btn.click(
             fn=process_schedule,
             inputs=[start_date, end_date, off_days, num_lessons, logo_file],
-            outputs=[summary_output, schedule_table, pdf_download, csv_download]
-        ).then(
-            lambda: [gr.update(visible=True), gr.update(visible=True)],
-            outputs=[pdf_download, csv_download]
+            outputs=[summary_output, schedule_table, pdf_download, pdf_download]
         )
         
         # Example section
         gr.Markdown(
             """
-            ### 💡 Tips:
-            - **Date Format**: Use YYYY-MM-DD format (e.g., 2025-07-01)
-            - **Off Days**: Select days you want to skip (weekends, holidays, etc.)
-            - **Lessons**: Adjust the number based on your course requirements
-            - **Logo**: Upload PNG, JPG, or other image formats for PDF branding
+            ### 💡 Usage Tips:
+            
+            **📅 Date Format**: Always use YYYY-MM-DD format
+            - ✅ Correct: `2025-07-01`
+            - ❌ Wrong: `07/01/2025` or `1 July 2025`
+            
+            **🚫 Off Days**: Select any days you want to skip
+            - Example: Skip weekends → Select "Saturday" and "Sunday"
+            - Example: Skip Friday prayers → Select "Friday"
+            
+            **📚 Lessons**: Adjust based on your course
+            - Online course with 90 videos → Set to 90
+            - Textbook with 20 chapters → Set to 20
             
-            ### 🎯 Example:
-            - Start: 2025-07-01, End: 2025-08-30
-            - Off Days: Friday, Sunday
+            **🖼️ Logo**: Upload any image to brand your PDF
+            - Supported: PNG, JPG, JPEG, GIF
+            - Recommended size: 200x100 pixels or similar ratio
+            
+            ### 🎯 Example Scenarios:
+            
+            **Scenario 1: Weekend Study**
+            - Start: 2025-07-01, End: 2025-12-31
+            - Off Days: Monday, Tuesday, Wednesday, Thursday, Friday
+            - Lessons: 50
+            - Result: Weekend-only schedule
+            
+            **Scenario 2: Intensive Course**
+            - Start: 2025-07-01, End: 2025-07-31
+            - Off Days: Sunday
             - Lessons: 90
-            - This creates a schedule with 1-2 lessons per available day
+            - Result: ~3-4 lessons per day
+            
+            **Scenario 3: Relaxed Learning**
+            - Start: 2025-07-01, End: 2025-12-31
+            - Off Days: Friday, Saturday, Sunday
+            - Lessons: 100
+            - Result: ~1 lesson per weekday
             """
         )
     
@@ -377,5 +406,5 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=False,
-        debug=False
+        debug=True
     )