Update app.py

app.py

@@ -46,7 +46,7 @@ def load_and_extract_text(file_path):
                 reader = csv.reader(f)
                 return '\n'.join(' '.join(row) for row in reader)
 
-        elif ext == 'txt':
+        elif ext == 'txt' or ext == 'py':  # ✅ Added .py support here
             with open(file_path, 'r', encoding='utf-8') as f:
                 return f.read()
 
@@ -61,7 +61,7 @@ def load_and_extract_text(file_path):
         return f"Error processing file: {e}"
 
 # File uploader
-uploaded_file = st.file_uploader("Upload a document or image", type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "jpg", "jpeg", "png"])
+uploaded_file = st.file_uploader("Upload a document or image", type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "jpg", "jpeg", "png", "py"])  # ✅ Include .py here
 
 # Question input
 question = st.text_input("💬 Enter your question:")