Rebase

README.md

@@ -5,8 +5,8 @@ colorFrom: red
 colorTo: gray
 sdk: gradio
 sdk_version: 3.33.1
-app_file: app.py
-pinned: false
+app_file: bot.py
+pinned: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py → bot.py

@@ -1,5 +1,5 @@
 import discord
-from discord import Option
+from discord import Option, commands
 import redis
 import os
 
@@ -108,18 +108,37 @@ async def image(ctx, id):
     embed.set_image(url=path)
     await ctx.respond(embed=embed)
 
+def confAutocomplete(self: discord.AutocompleteContext):
+        return ['low', 'medium', 'high']
+
+def get_conf_score(confidence: str):
+    match confidence:
+        case "low":
+            return 0
+        case "medium":
+            return 1
+        case "high":
+            return 2
+        case _:
+            return 0
+
 @bot.command(
         description="Label image observation",
         options=[
             Option(name="id", description="The ID of the image"),
             Option(name="label", description="The label for the image"),
-            Option(name="confidence", description="The confidence for the label"),
+            Option(name="confidence", description="Choose your confidence", autocomplete=confAutocomplete)
     ])
-async def label_image(ctx, id:int, label:str, confidence:int):
+async def label_image(ctx, id:int, label:str, confidence: str):
     label_cnt = generate_id(f'label:{id}')
     r.hset(f'label:{id}:{label_cnt}'.encode('utf-8'), b'label', label.encode('utf-8'))
     r.hset(f'label:{id}:{label_cnt}'.encode('utf-8'), b'author', ctx.author.mention.encode('utf-8'))
-    r.hset(f'label:{id}:{label_cnt}'.encode('utf-8'), b'confidence', ctx.author.mention.encode('utf-8'))
+    conf_score = get_conf_score(confidence)
+    r.hset(f'label:{id}:{label_cnt}'.encode('utf-8'), b'confidence', conf_score)
+
+    r.hset(f'image:{id}'.encode('utf-8'), b'label', label.encode('utf-8'))
+    r.hset(f'image:{id}'.encode('utf-8'), b'author', ctx.author.mention.encode('utf-8'))
+
 
     # Fetch the image data from Redis
     image_data_bytes = r.hgetall(f'image:{id}'.encode('utf-8'))
@@ -132,7 +151,8 @@ async def label_image(ctx, id:int, label:str, confidence:int):
     embed.add_field(name="Sensor", value=image_data['sensor'], inline=True)
     embed.add_field(name="Label", value=image_data['label'], inline=True)
     embed.add_field(name="Proposed by", value=image_data['author'], inline=True)
-    embed.set_image(url=image_data['path'])
+    path = f'https://gainforest-transparency-dashboard.s3.amazonaws.com/{image_data["awsCID"]}'
+    embed.set_image(url=path)
 
     improve_player_stats(ctx)