Update app.py

app.py

@@ -23,6 +23,8 @@ genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel(model_name="gemini-2.0-flash")  # Supports both doc & image queries
 
 # Helper: Extract text from various documents
+import re  # Add this at the top for LaTeX cleaning
+
 def load_and_extract_text(file_path):
     try:
         ext = file_path.split('.')[-1].lower()
@@ -54,6 +56,15 @@ def load_and_extract_text(file_path):
             prs = Presentation(file_path)
             return '\n'.join(shape.text for slide in prs.slides for shape in slide.shapes if hasattr(shape, 'text'))
 
+        elif ext == 'tex':
+            with open(file_path, 'r', encoding='utf-8') as f:
+                content = f.read()
+                # Remove LaTeX commands (basic approach)
+                text = re.sub(r'\\[a-zA-Z]+\*?(?:\[[^\]]*\])?(?:\{[^}]*\})*', '', content)
+                # Remove comments
+                text = re.sub(r'%.*', '', text)
+                return text.strip()
+
         else:
             return "Unsupported file format."
 
@@ -61,7 +72,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", "tex", "jpg", "jpeg", "png"])
 
 # Question input
 question = st.text_input("💬 Enter your question:")