Update app.py

app.py

@@ -118,17 +118,21 @@ def generate_notes_flowchart(text):
     return notes,img
 
 # =====================
-# FILE OCR
+# FILE OCR (Safe)
 # =====================
 def extract_text_from_file(file):
-    if file.name.endswith(".pdf"):
-        pages=convert_from_path(file.name)
-        text=""
-        for page in pages:
-            text+=pytesseract.image_to_string(page)+"\n"
-        return text
-    else:
-        return pytesseract.image_to_string(Image.open(file.name))
+    try:
+        if file.name.endswith(".pdf"):
+            pages=convert_from_path(file.name)
+            text=""
+            for page in pages:
+                text+=pytesseract.image_to_string(page)+"\n"
+            return text
+        else:
+            img = Image.open(file.name)
+            return pytesseract.image_to_string(img)
+    except Exception as e:
+        return f"Error reading file: {e}"
 
 # =====================
 # LINKS
@@ -147,11 +151,64 @@ custom_css="""
 body {background: linear-gradient(to bottom right, #90ee90, #ffb6c1);}
 """
 
+# =====================
+# UPLOAD PROCESS (Fixed)
+# =====================
+def process_file(file, options, lg):
+    text=""
+    mcq=""
+    prev=None
+    full=None
+    notes=""
+    flow=None
+
+    if file is None:
+        return "No file uploaded.","",None,None,"",None
+
+    # Extract text safely
+    try:
+        if file.name.endswith(".pdf"):
+            pages=convert_from_path(file.name)
+            for page in pages:
+                text+=pytesseract.image_to_string(page)+"\n"
+        else:
+            img = Image.open(file.name)
+            text = pytesseract.image_to_string(img)
+    except Exception as e:
+        text=f"Error extracting text: {e}"
+
+    # Generate MCQs
+    if "MCQs" in options and text.strip():
+        try:
+            mcq = generate_mcqs_from_text(text)
+        except Exception as e:
+            mcq=f"Error generating MCQs: {e}"
+
+    # Generate Video
+    if "Video" in options and text.strip():
+        try:
+            prev=create_video_simple(text,lg,preview=True)
+            full=create_video_simple(text,lg)
+        except Exception as e:
+            prev=None
+            full=None
+            text += f"\n[Video error: {e}]"
+
+    # Generate Notes & Flowchart
+    if "Notes" in options and text.strip():
+        try:
+            notes,flow = generate_notes_flowchart(text)
+        except Exception as e:
+            notes=""
+            flow=None
+            text += f"\n[Notes/Flowchart error: {e}]"
+
+    return text, mcq, prev, full, notes, flow
+
 # =====================
 # GRADIO UI
 # =====================
 with gr.Blocks() as demo:
-    # Inject CSS for older Gradio versions
     gr.HTML(f"<style>{custom_css}</style>")
     gr.Markdown("## 🌸 AI Study Wizard — Pink + Green Gradient")
 
@@ -208,20 +265,6 @@ with gr.Blocks() as demo:
         fv2=gr.Video(label="Full Video",width=640,height=360)
         notes_out=gr.Textbox(label="Notes",lines=12)
         flow_out=gr.Image(label="Flowchart",type="pil")
-        def process_file(file,options,lg):
-            text=extract_text_from_file(file)
-            mcq=""
-            prev=None
-            full=None
-            notes=""
-            flow=None
-            if "MCQs" in options: mcq=generate_mcqs_from_text(text)
-            if "Video" in options:
-                prev=create_video_simple(text,lg,preview=True)
-                full=create_video_simple(text,lg)
-            if "Notes" in options:
-                notes,flow=generate_notes_flowchart(text)
-            return text,mcq,prev,full,notes,flow
         btn=gr.Button("Process")
         btn.click(process_file,[f_in,opts,lang],[t_out,mc_out,pv2,fv2,notes_out,flow_out])