Update app.py

app.py

@@ -9,6 +9,7 @@ from pptx import Presentation
 from io import StringIO
 from bs4 import BeautifulSoup
 import re
+import json
 import google.generativeai as genai
 
 # Setup
@@ -72,6 +73,22 @@ def load_and_extract_text(file_path):
                     script_or_style.decompose()
                 return soup.get_text(separator='\n', strip=True)
 
+        elif ext == 'py':
+            with open(file_path, 'r', encoding='utf-8') as f:
+                return f.read()
+
+        elif ext == 'ipynb':
+            with open(file_path, 'r', encoding='utf-8') as f:
+                notebook = json.load(f)
+                cells = notebook.get("cells", [])
+                extracted = []
+                for cell in cells:
+                    if cell.get("cell_type") == "markdown":
+                        extracted.append("### Markdown:\n" + ''.join(cell.get("source", [])))
+                    elif cell.get("cell_type") == "code":
+                        extracted.append("### Code:\n" + ''.join(cell.get("source", [])))
+                return '\n'.join(extracted)
+
         else:
             return "Unsupported file format."
 
@@ -81,7 +98,10 @@ def load_and_extract_text(file_path):
 # File uploader
 uploaded_file = st.file_uploader(
     "Upload a document or image",
-    type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "tex", "html", "htm", "jpg", "jpeg", "png"]
+    type=[
+        "docx", "xlsx", "pdf", "csv", "txt", "pptx", "tex", "html", "htm",
+        "jpg", "jpeg", "png", "py", "ipynb"
+    ]
 )
 
 # Question input