Update app.py

app.py

@@ -8,15 +8,17 @@ import base64
 import io
 import openpyxl
 from datetime import datetime
+import httpx
 
 # Load environment variables from .env file
 load_dotenv()
 
-# Initialize Groq client
-client = Groq(
-    api_key=os.environ.get("GROQ_API_KEY", ""),
-    base_url="https://api.groq.com/openai/v1"
-)
+def create_groq_client():
+    api_key = os.environ.get("GROQ_API_KEY", "")
+    return httpx.Client(
+        base_url="https://api.groq.com/openai/v1",
+        headers={"Authorization": f"Bearer {api_key}"}
+    )
 
 def encode_image_to_base64(image_path):
     """Convert image to base64 string"""
@@ -47,12 +49,6 @@ def extract_invoice_details(image):
                 "Price/Rate": "40.00",  # Numeric only
                 "Quantity": "2",  # Numeric only
                 "Amount": "80.00"  # Numeric only
-            },
-            {
-                "Item Name": "Product 2",
-                "Price/Rate": "19.00",
-                "Quantity": "6",
-                "Amount": "114.00"
             }
         ],
         "Total Invoice Value": "2555.00"  # Numeric only, total amount
@@ -61,28 +57,32 @@ def extract_invoice_details(image):
     Provide ONLY the dictionary, no additional text or formatting."""
     
     # Make API call to Groq
-    response = client.chat.completions.create(
-        model="llama-3.2-90b-vision-preview",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "image_url",
-                        "image_url": {
-                            "url": f"data:image/png;base64,{base64_image}"
+    client = create_groq_client()
+    response = client.post(
+        "/chat/completions",
+        json={
+            "model": "llama-3.2-90b-vision-preview",
+            "messages": [
+                {
+                    "role": "user",
+                    "content": [
+                        {
+                            "type": "image_url",
+                            "image_url": {
+                                "url": f"data:image/png;base64,{base64_image}"
+                            }
+                        },
+                        {
+                            "type": "text",
+                            "text": prompt
                         }
-                    },
-                    {
-                        "type": "text",
-                        "text": prompt
-                    }
-                ]
-            }
-        ]
+                    ]
+                }
+            ]
+        }
     )
-    
-    return response.choices[0].message.content
+    response_data = response.json()
+    return response_data['choices'][0]['message']['content']
 
 def parse_response(response_text):
     """Parse the model's response into structured data"""
@@ -191,4 +191,4 @@ iface = gr.Interface(
 
 # Launch the application
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(server_name="0.0.0.0", server_port=7860)