Update app.py

app.py

@@ -38,9 +38,18 @@ def extract_equations_and_generate_commands(api_response):
         if not line:
             continue
             
-        # Check if line contains LaTeX math
-        if '\\' in line and not line.startswith('#'):
-            # Extract LaTeX content
+        # Check if line contains any equation-like content
+        if line and (
+            '\\' in line or      # LaTeX commands
+            '=' in line or       # Equations
+            '+' in line or       # Mathematical operations
+            '->' in line or      # Arrows
+            '→' in line or       # Unicode arrow
+            '_' in line or       # Subscripts
+            '^' in line or       # Superscripts
+            any(char.isdigit() for char in line)  # Contains numbers
+        ) and not line.startswith('#'):
+            # Extract content
             latex_content = line
             
             # Clean up common markdown artifacts
@@ -143,21 +152,25 @@ def image_to_latex(image):
                 {
                     "parts": [
                         {
-                            "text": """Please analyze this image and convert any mathematical equations, formulas, or mathematical text you find into proper LaTeX format. 
+                            "text": """Please analyze this image and convert any equations, formulas, or expressions you find into proper LaTeX format. This includes mathematical equations, chemical equations, physics formulas, or any other scientific notation.
 
 Instructions:
-- Extract each mathematical equation or formula separately
-- Use proper LaTeX syntax for mathematical expressions
+- Extract each equation or formula separately
+- Use proper LaTeX syntax for all expressions
+- For chemical equations: use subscripts {}, superscripts {}, and \\rightarrow for arrows
+- For mathematical equations: use proper LaTeX math notation
 - Output raw LaTeX without any $ delimiters
 - Put each equation on a separate line
 - Do not include any markdown formatting or explanatory text
 - Only output the raw LaTeX code for each equation
 - If there are multiple equations, list them one per line
+- Include ALL text content that looks like equations or formulas, even if handwritten
 
 Example output format:
 \\frac{a}{b} = c
-x^2 + y^2 = z^2
-\\int_0^1 x dx = \\frac{1}{2}"""
+C_6H_{12}O_6 + 6O_2 \\rightarrow 6CO_2 + 6H_2O
+F = ma
+x^2 + y^2 = z^2"""
                         },
                         {
                             "inline_data": {
@@ -184,7 +197,17 @@ x^2 + y^2 = z^2
                 equations = []
                 for line in latex_output.split('\n'):
                     line = line.strip()
-                    if line and '\\' in line:
+                    # More flexible detection - look for common equation patterns
+                    if line and (
+                        '\\' in line or  # LaTeX commands
+                        '=' in line or   # Equations
+                        '+' in line or   # Mathematical operations
+                        '->' in line or  # Arrows (could be chemical)
+                        '→' in line or   # Unicode arrow
+                        '_' in line or   # Subscripts
+                        '^' in line or   # Superscripts
+                        any(char.isdigit() for char in line)  # Contains numbers
+                    ):
                         equations.append(line)
                 
                 if not equations:
@@ -246,21 +269,23 @@ def create_app():
             """
             # 📸 Image to maths2svg Command Generator
             
-            Upload an image containing mathematical equations and get ready-to-use `maths2svg` commands 
-            with intelligent filename suggestions.
+            Upload an image containing mathematical equations, chemical equations, physics formulas, 
+            or any scientific expressions and get ready-to-use `maths2svg` commands with intelligent filename suggestions.
+            
+            **Supported content:**
+            - Mathematical equations and formulas
+            - Chemical equations and reactions  
+            - Physics formulas and expressions
+            - Handwritten or printed equations
+            - Multiple equations in one image
             
             **Features:**
-            - Extracts multiple equations from images
+            - Extracts equations from images (including handwritten)
             - Generates proper LaTeX syntax
             - Creates maths2svg CLI commands
             - Suggests meaningful filenames using AI
             - Falls back to serial numbering if needed
             
-            **Supported content:**
-            - Mathematical equations and formulas
-            - Mathematical symbols and expressions
-            - Multiple equations in one image
-            
             **Note:** Make sure to set your `GEMINI_API_KEY` environment variable before running this app.
             """
         )
@@ -268,7 +293,7 @@ def create_app():
         with gr.Row():
             with gr.Column(scale=1):
                 image_input = gr.Image(
-                    label="Upload Image with Math Equations",
+                    label="Upload Image with Equations/Formulas",
                     type="pil",
                     height=400
                 )
@@ -293,9 +318,10 @@ def create_app():
         gr.Markdown(
             """
             - Use clear, high-contrast images
-            - Ensure mathematical expressions are clearly visible
+            - Ensure equations are clearly visible and well-lit
             - Multiple equations in one image are supported
-            - Handwritten or printed equations both work
+            - Works with handwritten, printed, or digital equations
+            - Supports mathematical equations, chemical formulas, physics expressions
             - The app will generate meaningful filenames automatically
             """
         )