Create app.py

app.py

@@ -0,0 +1,97 @@
+import os
+import json
+import base64
+from flask import Flask, request, jsonify
+from groq import Groq
+from werkzeug.utils import secure_filename
+
+app = Flask(__name__)
+app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max file size
+app.config['UPLOAD_FOLDER'] = 'uploads'
+
+# Initialize Groq client with your API key
+client = Groq(api_key="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")
+
+# Ensure upload folder exists
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+
+def encode_image_to_base64(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
+
+def extract_business_card_info(image_base64):
+    completion = client.chat.completions.create(
+        model="llama-3.2-11b-vision-preview",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": """Extract info STRICTLY in this JSON format ONLY:
+{
+  "businessCard": {
+    "companyName": "",
+    "email": "",
+    "contactNumber": "",
+    "website": "",
+    "address": ""
+  }
+}
+NO TEXT BEFORE/AFTER THE JSON. NO MARKDOWN. ONLY VALID JSON."""
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": f"data:image/jpeg;base64,{image_base64}"
+                        }
+                    }
+                ]
+            }
+        ],
+        temperature=0.3,
+        max_tokens=1024,
+        top_p=1,
+        stream=False,
+    )
+    return completion.choices[0].message.content
+
+@app.route('/extract', methods=['POST'])
+def extract_info():
+    if 'image' not in request.files:
+        return jsonify({'error': 'No image file provided'}), 400
+    
+    file = request.files['image']
+    if file.filename == '':
+        return jsonify({'error': 'No selected file'}), 400
+    
+    if file:
+        filename = secure_filename(file.filename)
+        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+        file.save(filepath)
+        
+        try:
+            # Convert image to base64
+            image_base64 = encode_image_to_base64(filepath)
+            
+            # Extract information
+            json_data = extract_business_card_info(image_base64)
+            
+            # Clean and parse JSON
+            json_start = json_data.find('{')
+            json_end = json_data.rfind('}') + 1
+            clean_json = json_data[json_start:json_end]
+            parsed_json = json.loads(clean_json)
+            
+            # Clean up the uploaded file
+            os.remove(filepath)
+            
+            return jsonify(parsed_json)
+            
+        except Exception as e:
+            if os.path.exists(filepath):
+                os.remove(filepath)
+            return jsonify({'error': str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)