Update app.py

app.py

@@ -17,7 +17,7 @@ class CalendarWrappedAPI:
         self.llm = OpenAI(model="gpt-4", api_key=os.getenv('OPENAI_API_KEY'))
         self.connections = {}
         
-    def initiate_connection(self, entity_id, redirect_url=None):
+    def initiate_connection(self, entity_id):
         """Initialize connection using entity_id (username)"""
 
         try:
@@ -31,7 +31,9 @@ class CalendarWrappedAPI:
                 'status': connection_request.connectionStatus,
                 'redirect_url': connection_request.redirectUrl
             }
-            
+            return {
+                'redirect_url':f'<a href="{connection_request.redirectUrl}" target="_blank">{connection_request.redirectUrl}</a>'
+            }
             # Wait for random time between 60-100 seconds
             wait_time = random.randint(60, 100)
             print(f"Waiting for {wait_time} seconds before checking connection status...")
@@ -102,8 +104,8 @@ class CalendarWrappedAPI:
 def create_gradio_api():
     api = CalendarWrappedAPI()
     
-    def handle_connection(entity_id, redirect_url):
-        result = api.initiate_connection(entity_id, redirect_url)
+    def handle_connection(entity_id):
+        result = api.initiate_connection(entity_id)
         return {
             **result,
             'message': f"Connection initiated. Waited for {result.get('wait_time', 0)} seconds."
@@ -119,8 +121,7 @@ def create_gradio_api():
     with gr.Blocks() as demo:
         gr.Markdown("""
         # Google Calendar Wrapped Generator
-        **Note**: After initiating connection, check the console/terminal for the redirect URL.
-        The status will be checked automatically after 60-100 seconds.
+        **Note**: After initiating connection, check the connection status and generate your wrapped summary.
         """)
         
         with gr.Tab("Connection"):
@@ -128,16 +129,12 @@ def create_gradio_api():
                 label="Entity ID (Username)",
                 placeholder="Enter your username as entity ID"
             )
-            redirect_url_input = gr.Textbox(
-                label="Redirect URL",
-                placeholder="https://yourwebsite.com/connection/success"
-            )
             connect_btn = gr.Button("Initialize Connection")
             connection_output = gr.HTML(label="Connection Status")
             
             connect_btn.click(
                 fn=handle_connection,
-                inputs=[entity_id_input, redirect_url_input],
+                inputs=entity_id_input,
                 outputs=connection_output
             )