Update helpers/llm_helper.py

helpers/llm_helper.py

@@ -8,13 +8,16 @@ def analyze_image_file(image_file, model, user_prompt):
     # gets image bytes using helper function
     image_bytes = get_image_bytes(image_file)
 
-    # calls the llava model using Ollama SDK
-    stream = generate(model=model, 
-            prompt=user_prompt, 
-            images=[image_bytes], 
-            stream=True)
+    # API call to remote Ollama server (update with actual URL)
+    url = 'https://502f-2401-4900-882d-894-a2-f66f-ba2b-969f.ngrok-free.app'
+    data = {
+        'model': model,
+        'prompt': user_prompt,
+        'image_bytes': image_bytes
+    }
 
-    return stream
+    response = requests.post(url, json=data)
+    return response.json()
 
 # handles stream response back from LLM
 def stream_parser(stream):