Update app.py

app.py

@@ -1,120 +1,76 @@
 import gradio as gr
-from collections import defaultdict
-import random
-import pandas as pd
-import os
-import cloudinary
-import cloudinary.uploader
-from cloudinary.utils import cloudinary_url
-import io
-import time
 import requests
+import json
+import os
+from datasets import load_dataset
 
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
 
+db_password = os.environ["db_password"]
 
-# Configuration
-cloudinary.config(
-    cloud_name = os.environ["cloud_name"],
-    api_key = os.environ["api_key"] ,
-    api_secret = os.environ["api_secret"] ,
-    secure=True
-)
+uri = f"mongodb+srv://tuna:{db_password}@tuna.ixw4ff2.mongodb.net/?appName=tuna"
 
-from datetime import datetime
-import pytz
+# Create a new client and connect to the server
+client = MongoClient(uri, server_api=ServerApi('1'))
 
-# Define the GMT+7 timezone
-gmt_plus7 = pytz.timezone('Etc/GMT-7')  # 'Etc/GMT-7' represents GMT+7
+database = client['valid_image_caption']
+collection = database["log_valid"]
 
+ds = load_dataset("anhdt-dsai-02/test_image_dataset_1_2_3_4", token = os.environ['token_huggingface'] )
 
+def get_similar_captions(caption):
 
-def authenticate(user_id):
+    url = "https://anhdt-dsai-02-caption-retrieval.hf.space/retrieval"
+    params = {"caption": caption}
 
-    url = "https://intern-api.imtaai.com/api/subnets/2/authenticate"
-    headers = {
-        "Content-Type": "application/json",
-        "Accept": "application/json",
-        "X-Public-Api-Key": os.environ['ADMIN']
-        }
-    payload = { "token": user_id }
-    response = requests.post(url, json=payload, headers=headers)
+    try:
+        response = requests.post(url, params=params, headers={"accept": "application/json"})
+        response.raise_for_status()  # Raises an HTTPError for bad responses (4xx and 5xx)
+        data = response.json()  # Assuming response is JSON
 
-    return response.status_code == 200
+    except requests.exceptions.RequestException as e:
+        print(f"Request failed: {e}")
+    except ValueError:
+        print("Failed to parse JSON response")
 
-def login(username, state, package):
-    state[0] = username
-    #package[0] = get_next_package(user_id=username)
-    """
-    #temp
-    gr.Info("Login successfully. Welcome!")
-    return f"Welcome, {username}!", gr.update(visible=False), gr.update(visible=True)
-    #temp
-    """
-    
-    # Authenticate user
-    if authenticate(username):
-        #user_sessions[username] = True
-        gr.Info("Login successfully. Welcome!")
-        return f"Welcome, {username}!", gr.update(visible=False), gr.update(visible=True)
-    else:
-        raise gr.Error("Token ID is invalid! Try again!")
-        return "Invalid Token ID", gr.update(visible=True), gr.update(visible=False)
+    return data
+
+
+
+index = 0
+
+def create_question():
+    global index
+    print(index)
+    caption = ds['train'][index]["caption"] # delete [0]
+    caption_lst = get_similar_captions(caption)
+    #id_lst = [pair['id'] for pair in caption_lst]
+
+    id_lst = [int(pair['id']/10) for pair in caption_lst]
 
-def logout(username):
-    # Log out user and reset session
-    if username in user_sessions:
-        del user_sessions[username]
-    gr.Warning("You need to login to access this subnet and do task ⛔️!", duration=5)
-    return "Logged out. Please log in again.", gr.update(visible=True), gr.update(visible=False)
-
-def get_image_bytes(image):
-    if image is None:
-        return "No image uploaded!"
-
-    # Convert the image to bytes
-    img_byte_arr = io.BytesIO()
-    image.save(img_byte_arr, format='JPEG')  # Save as PNG or desired format
-    img_bytes = img_byte_arr.getvalue()
-
-    return img_bytes
-
-
-# Define a function to handle image and text caption
-def process_image_and_caption(username_input, image, caption):
-    #check image 
-    if image is None:
-        gr.Warning("No image uploaded!", duration=5)
-        return "No image uploaded!", image, caption
-
-    # check caption
-    if not caption.strip():
-        gr.Warning("No caption uploaded!", duration=5)
-        return "No caption uploaded!", image, caption
-    # Get current time in GMT+7
-    current_time = datetime.now(gmt_plus7)
-
-    # Print the time
-    time = current_time.strftime('-%Y-%m-%d-%H-%M-%S')
-    name = username_input + time
-    # Upload an image
-    img_bytes = get_image_bytes(image)
-    upload_image_result = cloudinary.uploader.upload(img_bytes, public_id = name, asset_folder = "image_for_captioning")
-
-
-    # Upload caption
-    # Convert to bytes
-    caption_bytes = caption.encode('utf-8')
-    # Create a byte array buffer (like a binary file stream)
-    byte_buffer = io.BytesIO(caption_bytes)
-    upload_caption_result = cloudinary.uploader.upload(byte_buffer, public_id = name, asset_folder = "caption", resource_type = "raw")
-    send_score(username_input, 0.1)
-    gr.Info("Submit successfully")
-    return f'You just upload an Image with Caption: "{caption}" \nIntime {time}', None, None
-
-def send_score(user_id, score):
+    images = [ds["train"][id]['image'] for id in id_lst]
+
+    index += 1
+
+    return caption, images, id_lst
+
+def on_select(evt: gr.SelectData):
+    idx = evt.index
+    return idx
+
+# Function to handle user selection
+def process_answer(user_id, caption, selected_image, ids):
+
+    collection.insert_one({ "used_id" : user_id, "caption": caption, "ids":  ids, "choice": selected_image})
+    send_score(user_id, 0.1)
+
+    return create_question()
+
+def send_score(user_id, score = 0.1):
     max_retries = 10
     while max_retries > 0:
-        url = "https://intern-api.imtaai.com/api/subnets/2/grade"
+        url = os.environ['api_url'] + "grade"
 
         payload = {
             "token": user_id,
@@ -136,50 +92,86 @@ def send_score(user_id, score):
         max_retries -= 1
     return False
 
-# Define the Gradio interface
+def authenticate(user_id):
+
+    url = os.environ['api_url'] + "authenticate"
+    headers = {
+        "Content-Type": "application/json",
+        "Accept": "application/json",
+        "X-Public-Api-Key": os.environ['ADMIN']
+        }
+    payload = { "token": user_id }
+    response = requests.post(url, json=payload, headers=headers)
+
+    return response.status_code == 200
+
+
+
+def login(username):
+    #state[0] = username
+    #package[0] = get_next_package(user_id=username)
+    """
+    #temp
+    gr.Info("Login successfully. Welcome!")
+    return f"Welcome, {username}!", gr.update(visible=False), gr.update(visible=True)
+    #temp
+    """
+
+    # Authenticate user
+    if authenticate(username):
+        #user_sessions[username] = True
+        gr.Info("Login successfully. Welcome!")
+        return f"Welcome, {username}!", gr.update(visible=False), gr.update(visible=True), *create_question()
+    else:
+        raise gr.Error("Token ID is invalid! Try again!")
+        return "Invalid Token ID", gr.update(visible=True), gr.update(visible=False)
+
+# Gradio UI
 with gr.Blocks() as demo:
-    state = gr.State([None])
-    package = gr.State([None])
-    # Login section
     with gr.Column(visible=True) as login_section:
         username_input = gr.Textbox(placeholder="Enter your token", label="Token ID", type="password")
         login_button = gr.Button("Login")
         login_output = gr.Textbox(label="Login Status", interactive=False)
 
-     # Upload section  (initially hidden)
+
+    # Upload section  (initially hidden)
     with gr.Column(visible=False) as upload_section:
-        gr.Markdown("""### Upload hình ảnh (image) kèm chú thích (captioning).
-        - Chú ý: Hình ảnh phải rõ nét, có nội dung; chú thích phải miêu tả chính xác nội dung của hình ảnh.
-        <div style='color: red; font-size: 18px;'>Không đăng tải các nội dung cá nhân, nhạy cảm vì dữ liệu có thể được public cho cộng đồng trong lương lai</div>""")
-        with gr.Row():
-            
-            image_input = gr.Image(type="pil", label="Upload Image")
-            caption_input = gr.Textbox(label="Enter Caption")
+        caption = gr.State()
+        images = gr.State()
+        id_lst = gr.State()
+        # Store the selected image path
+        selected_image = gr.State()
 
-        submit_button = gr.Button("Submit")
-        output = gr.Textbox(label="Output")
 
-    # Button functions
+        #caption.value, images.value, id_lst.value = create_question()
 
+        gr.Markdown("**Đâu là hình ảnh phù hợp với caption sau:**")
+        markdown = gr.Markdown(caption.value)
 
-    submit_button.click(
-        process_image_and_caption,
-        inputs=[username_input, image_input, caption_input],
-        outputs=[output, image_input, caption_input]
-    )
-    caption_input.submit(
-        process_image_and_caption,
-        inputs=[username_input, image_input, caption_input],
-        outputs=[output, image_input, caption_input]
-    )
 
-    
+
+        # Display images in a Gallery
+        gallery = gr.Gallery(value=images.value, label="Ấn vào hình ảnh để chọn sau đó Submit", columns = 4)
+
+        # Use the Gallery's select event to update the selected image
+        gallery.select(fn=on_select, outputs=selected_image)
+
+        # Submit button
+        submit_button = gr.Button("Submit")
+
+        # Output message
+        output_message = gr.Textbox(label="Result")
+
+    # Action in Login Section
     username_input.submit(
-        login, inputs=[username_input, state, package], outputs=[login_output, login_section, upload_section] #, translation_section, en_input, vi_input]
+        login, inputs=[username_input], outputs=[login_output, login_section, upload_section,  markdown, gallery, id_lst] #, translation_section, en_input, vi_input]
     )
     login_button.click(
-        login, inputs=[username_input, state, package], outputs=[login_output, login_section, upload_section] #, translation_section, en_input, vi_input]
+        login, inputs=[username_input], outputs=[login_output, login_section, upload_section,  markdown, gallery, id_lst] #, translation_section, en_input, vi_input]
     )
 
+    # Link the submit button to the processing function
 
-demo.launch(debug=True)
+    submit_button.click(fn=process_answer, inputs=[username_input, caption, selected_image, id_lst], outputs=[markdown, gallery, id_lst])
+# Launch the app
+demo.launch(debug = True)