Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
-import requests
 import base64
+import requests
 import os
+import random
+import time
 
 # Function to convert image to base64
 def convert_image_to_base64(image):
@@ -19,8 +21,8 @@ def generate_caption(encoded_image):
         "Content-Type": "application/json"
     }
 
-    # Print base64 to debug (make sure it's different for each image upload)
-    print(f"Base64 encoded image: {encoded_image[:100]}...")  # Print only first 100 characters for brevity
+    # Add a random string to the URL to prevent caching issues
+    random_string = str(random.randint(100000, 999999))
 
     payload = {
         "model": "Qwen/Qwen2-VL-7B-Instruct",
@@ -37,7 +39,7 @@ def generate_caption(encoded_image):
         "image": {
             "type": "image_url",
             "image_url": {
-                "url": f"data:image/png;base64,{encoded_image}"
+                "url": f"data:image/png;base64,{encoded_image}?{random_string}"  # Added random query to avoid cache
             }
         },
         "temperature": 0
@@ -62,7 +64,7 @@ def main():
     st.markdown("Upload an image and let the AI generate a detailed caption for it.")
 
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-    
+
     if uploaded_file:
         # Display the uploaded image
         st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
@@ -71,7 +73,7 @@ def main():
         if st.button("Generate Caption"):
             with st.spinner("Generating caption..."):
                 encoded_image = convert_image_to_base64(uploaded_file)
-                
+
                 # Debugging: print out the base64 string length to verify if it's changing
                 st.write(f"Encoded image length: {len(encoded_image)} characters")