Create app.py

app.py

@@ -0,0 +1,79 @@
+import gradio as gr
+import google.generativeai as genai
+from PIL import Image
+import os
+import re
+
+# Direct API key placement
+API_KEY = 'AIzaSyBMZrhMXRpQKp7M-JcN2Qk73afeta5Mv5Y'
+
+def extract_malayalam_letters(image):
+    """
+    Extract ONLY Malayalam letters from an image using Gemini 2.0 Flash
+    
+    Args:
+        image (PIL.Image): Uploaded image
+    
+    Returns:
+        str: Extracted Malayalam letters
+    """
+    # Validate input
+    if image is None:
+        return "Please upload an image first."
+    
+    try:
+        # Configure the Gemini API 
+        genai.configure(api_key=API_KEY)
+        
+        # Use Gemini 2.0 Flash model
+        model = genai.GenerativeModel('gemini-2.0-flash')
+        
+        # Prompt to focus on Malayalam letters
+        response = model.generate_content(
+            [
+                "CRITICAL INSTRUCTIONS: "
+                "This is a handwritten image. "
+                "Extract ONLY Malayalam letters. "
+                "Rules for extraction: "
+                "1. Return ONLY pure Malayalam Unicode characters "
+                "2. Remove ALL numbers and English letters "
+                "3. Preserve original order "
+                "4. Do not translate or modify text", 
+                image
+            ],
+            generation_config=genai.types.GenerationConfig(
+                temperature=0.1,  # Ultra-low temperature for precise extraction
+                max_output_tokens=300  # Adjust based on expected text length
+            )
+        )
+        
+        # Extract text from response
+        extracted_text = response.text
+        
+        # Filter to keep only Malayalam letters
+        malayalam_letters = ''.join(
+            char for char in extracted_text 
+            if '\u0D00' <= char <= '\u0D7F'
+        )
+        
+        # Validate Malayalam letters extraction
+        if not malayalam_letters:
+            return "No Malayalam letters detected. Please check the image quality."
+        
+        return malayalam_letters
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
+# Create Gradio Interface
+demo = gr.Interface(
+    fn=extract_malayalam_letters,
+    inputs=gr.Image(type="pil", label="Upload Image with Malayalam Text"),
+    outputs=gr.Textbox(label="Extracted Malayalam Letters", lines=10),
+    title="Malayalam Letters Extractor",
+    description="Upload an image to extract ONLY Malayalam letters."
+)
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()