Update app.py

app.py

@@ -767,45 +767,69 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                     encoded_image = base64.b64encode(buffer.getvalue()).decode("utf-8")
                 return encoded_image
 
+            def url_to_base64(image_url):
+                try:
+                    # Download the image from the URL
+                    response = requests.get(image_url)
+                    response.raise_for_status()
+            
+                    # Convert the image content to base64
+                    base64_data = base64.b64encode(response.content).decode('utf-8')
+                    
+                    return base64_data
+                except Exception as e:
+                    print(f"Error: {e}")
+                    return None
+
+
             def vision():
-                with open("image.png", "wb") as file1_write:
-                    file1_write.write(image_data)
-        
-                with open("image.png", "rb") as file1_read:
-                    file_content = file1_read.read()
+                # with open("image.png", "wb") as file1_write:
+                #     file1_write.write(image_data)
+                
+                # Example usage
+                # image_url = 'https://example.com/path/to/image.jpg'
+                # base64_data = url_to_base64(image_url)
         
-                image = Image.open(io.BytesIO(file_content))
+                # image = Image.open(io.BytesIO(file_content))
             
-                base64_image_str = encode_image(image)
-                    
-                payload = {
-                    "content": [
-                        {
-                            "prompt": "What's this image about? or What does this image contains?",
-                            "image": base64_image_str,
+                # base64_image_str = encode_image(image)
+
+                if image_data:
+                    with io.BytesIO(base64.b64decode(image_data)) as buffer:
+                        image = Image.open(buffer)
+                        base64_image_str = encode_image(image)
+                        
+                        payload = {
+                            "content": [
+                                {
+                                    "prompt": "What's this image about? or What does this image contains?",
+                                    "image": base64_image_str,
+                                }
+                            ],
+                            "token": vis_auth,
                         }
-                    ],
-                    "token": vis_auth,
-                }
-            
-                url = vis_url
-                headers = {"Content-Type": "application/json"}
-            
-                response = requests.post(url, headers=headers, data=json.dumps(payload))
-                results = response.json()
-                results = results["result"]
-            
-                answer_index = results.find("Answer:")
-            
-                if answer_index != -1:
-                    try:
-                        result_text = results[answer_index + len("Answer:"):].strip()
-                        print(result_text)
-                        return result_text
-                    except:
-                        pass
+                    
+                        url = vis_url
+                        headers = {"Content-Type": "application/json"}
+                    
+                        response = requests.post(url, headers=headers, data=json.dumps(payload))
+                        results = response.json()
+                        results = results["result"]
+                    
+                        answer_index = results.find("Answer:")
+                    
+                        if answer_index != -1:
+                            try:
+                                result_text = results[answer_index + len("Answer:"):].strip()
+                                print(result_text)
+                                return result_text
+                            except:
+                                pass
                 else:
                     return "Answer: not found in the string."
+                else:
+                    print("Error: Image data is not available.")
+                    return None
 
             if "vision" in UserInput.lower():
                 vision()