Update main.py

main.py

@@ -378,30 +378,19 @@ def process_image():
         try:
             model = configure_gemini(api_key)
             
-            # Upload image file to Gemini
-            logging.info(f"Uploading image file: {file.filename}")
-            uploaded_file = genai.upload_file(file_path)
+            logging.info(f"Processing image file: {file.filename}")
             
-            # Wait for file to be processed
-            while uploaded_file.state.name == "PROCESSING":
-                time.sleep(1)
-                uploaded_file = genai.get_file(uploaded_file.name)
-            
-            if uploaded_file.state.name == "FAILED":
-                raise ValueError("Image processing failed")
-            
-            logging.info("Processing image for transaction extraction")
+            # Read image file as bytes
+            import PIL.Image
+            img = PIL.Image.open(file_path)
             
             # Generate content with the image and prompt
-            response = model.generate_content([uploaded_file, PROMPT])
+            response = model.generate_content([PROMPT, img])
             
             # Parse the response
             result = extract_json_from_response(response.text)
             transactions = result.get('transactions', [])
             
-            # Clean up the uploaded file from Gemini
-            genai.delete_file(uploaded_file.name)
-            
             return jsonify({'transactions': transactions})
             
         finally: