update auth.py

src/auth.py

@@ -25,38 +25,7 @@ class MALAuth:
         self.auth_code = None
         self.error = None
 
-    class OAuthHandler(BaseHTTPRequestHandler):
-        auth_code = None
-        error = None
-
-        def do_GET(self):
-            parsed = urlparse(self.path)
-            params = parse_qs(parsed.query)
-            if "code" in params:
-                MALAuth.OAuthHandler.auth_code = params["code"][0]
-                self.send_response(200)
-                self.end_headers()
-                self.wfile.write(b"Authorization successful! Return to the app.")
-            elif "error" in params:
-                MALAuth.OAuthHandler.error = params["error"][0]
-                self.send_response(400)
-                self.end_headers()
-                self.wfile.write(b"Authorization failed. Check your settings.")
-            else:
-                self.send_response(400)
-                self.end_headers()
-                self.wfile.write(b"Invalid request")
-
-    def run_server(self):
-        server = HTTPServer(('localhost', PORT), self.OAuthHandler)
-        server.timeout = 120
-        server.handle_request()
-
     def start_oauth_flow(self):
-        server_thread = threading.Thread(target=self.run_server)
-        server_thread.daemon = True
-        server_thread.start()
-
         auth_url = (
             "https://myanimelist.net/v1/oauth2/authorize?"
             f"response_type=code&"
@@ -64,15 +33,60 @@ class MALAuth:
             f"code_challenge={self.code_challenge}&"
             f"redirect_uri={REDIRECT_URI}"
         )
-        webbrowser.open(auth_url)
+        st.markdown(f"[Click here to authenticate with MyAnimeList]({auth_url})", unsafe_allow_html=True)
 
-        start_time = time.time()
-        while not self.OAuthHandler.auth_code and not self.OAuthHandler.error:
-            if time.time() - start_time > 120:
-                return None, "Authorization timed out"
-            time.sleep(0.5)
+        # Streamlit Community Cloud: handle redirect via query params
+        query_params = st.experimental_get_query_params()
+        if "code" in query_params:
+            self.auth_code = query_params["code"][0]
+            return self.auth_code, None
+        elif "error" in query_params:
+            self.error = query_params["error"][0]
+            return None, self.error
 
-        return self.OAuthHandler.auth_code, self.OAuthHandler.error
+        # Local: try to run a local HTTP server if running on localhost
+        if "localhost" in REDIRECT_URI or "127.0.0.1" in REDIRECT_URI:
+            try:
+                from http.server import HTTPServer, BaseHTTPRequestHandler
+                import threading
+                import time
+                class OAuthHandler(BaseHTTPRequestHandler):
+                    auth_code = None
+                    error = None
+                    def do_GET(self):
+                        from urllib.parse import urlparse, parse_qs
+                        parsed = urlparse(self.path)
+                        params = parse_qs(parsed.query)
+                        if "code" in params:
+                            OAuthHandler.auth_code = params["code"][0]
+                            self.send_response(200)
+                            self.end_headers()
+                            self.wfile.write(b"Authorization successful! Return to the app.")
+                        elif "error" in params:
+                            OAuthHandler.error = params["error"][0]
+                            self.send_response(400)
+                            self.end_headers()
+                            self.wfile.write(b"Authorization failed. Check your settings.")
+                        else:
+                            self.send_response(400)
+                            self.end_headers()
+                            self.wfile.write(b"Invalid request")
+                server = HTTPServer(('localhost', PORT), OAuthHandler)
+                server.timeout = 120
+                server_thread = threading.Thread(target=server.handle_request)
+                server_thread.daemon = True
+                server_thread.start()
+                import webbrowser
+                webbrowser.open(auth_url)
+                start_time = time.time()
+                while not OAuthHandler.auth_code and not OAuthHandler.error:
+                    if time.time() - start_time > 120:
+                        return None, "Authorization timed out"
+                    time.sleep(0.5)
+                return OAuthHandler.auth_code, OAuthHandler.error
+            except Exception as e:
+                return None, f"Local OAuth server failed: {e}"
+        return None, None
 
     def get_access_token(self, auth_code):
         token_url = "https://myanimelist.net/v1/oauth2/token"