Update app.py

app.py

@@ -9,7 +9,6 @@ from pptx import Presentation
 from io import StringIO
 from bs4 import BeautifulSoup
 import re
-import json
 import google.generativeai as genai
 
 # Setup
@@ -73,22 +72,6 @@ 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()  # We just read the Python code content
-
-        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."
 
@@ -98,10 +81,7 @@ 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", "py", "ipynb"
-    ]
+    type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "tex", "html", "htm", "jpg", "jpeg", "png"]
 )
 
 # Question input
@@ -146,11 +126,9 @@ if uploaded_file and question:
         else:
             with st.spinner("Processing document..."):
                 try:
-                    # Hide the extracted content
-                    st.write("Processing your query based on the uploaded content...")
                     prompt = f"Context: {extracted_text}\n\nQuestion: {question}\n\nAnswer:"
                     response = model.generate_content(prompt)
                     st.success("💡 Answer:")
                     st.write(response.text)
                 except Exception as e:
-                    st.error(f"Error generating answer: {e}")
+                    st.error(f"Error generating answer: {e}")