Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import gradio.components as comp
 import discord
 import os
 import requests
@@ -57,17 +56,17 @@ async def extract_messages(token, save_dir):
     # Logout of Discord
     await client.logout()
 
-    # Return number of messages and links extracted as a tuple
+    # Return number of messages and links extracted
     return len(messages), len(links)
 
 # Gradio interface
 def interface():
     # Input fields
-    token = comp.Textbox(label="Discord User Token")
-    save_dir = comp.Textbox(label="Save Directory")
+    token = gr.inputs.Textbox(label="Discord User Token")
+    save_dir = gr.inputs.Textbox(label="Save Directory")
 
     # Output field
-    output = gradio.components.Textbox()
+    output = gr.outputs.Text()
 
     # Function to extract messages and update progress
     async def extract_messages_with_progress(token, save_dir):
@@ -76,15 +75,14 @@ def interface():
 
         # Extract messages
         loop = asyncio.get_running_loop()
-        results = await loop.run_in_executor(None, extract_messages, token, save_dir)
-        num_messages, num_links = results
+        num_messages, num_links = await loop.run_in_executor(None, extract_messages, token, save_dir)
 
         # Update progress bar
         progress_bar.update(50)
         progress_bar.set_description("Extracting Links")
         progress_bar.update(50)
 
-        # Return output as a string
+        # Return output
         output_text = f"Messages Extracted: {num_messages}\nLinks Extracted: {num_links}"
         return output_text