Upload 3 files

Dockerfile

@@ -21,8 +21,10 @@ WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
+# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip install https://gradio-pypi-previews.s3.amazonaws.com/b176db9d6e1bffd36f5b2ecd54b425e4cec3dab8/gradio-5.33.0-py3-none-any.whl
+RUN pip install gradio[mcp]
+RUN pip install python-dotenv
 RUN pip install playwright
 RUN playwright install --with-deps
 

app.py

@@ -8,28 +8,35 @@ if os.getenv('ENVIRONMENT')=="PROD":
     print("installing playwright firefox")
     subprocess.run(["playwright","install","firefox"])
 
-def fetch_homework(date:str='today') -> str:
+
+def fetch_homework(date:str,request: gr.Request) -> str:
     """
     description:
         fetch the homeworks.
     Args: 
-        date: any string, default "today"
+        date: (str) set it to "today" unless explicitly specified otherwise
     Returns:
         The string describing the homeworks    
     """
-    return get_homework()
+    # print(str(dict(request.headers)["token"]))
+    token = str(dict(request.headers)["token"])
+
+    return get_homework(token)
 
 
-def fetch_timetable(date:str='today') -> str:
+def fetch_timetable(date:str,request: gr.Request) -> str:
     """
     description:
         fetch the timetable
     Args: 
-        date: any string, default "today"
+        date: (str) set it to "today" unless explicitly specified otherwise
     Returns:
         The string describing the timetable    
     """
-    return get_timetable()
+    # print(str(dict(request.headers)["token"]))
+    token = str(dict(request.headers)["token"])
+    return get_timetable(token)
+
 
 
 title="<h2>Gradio MCP Hackathon: fake-app-scraper</h2>"
@@ -58,7 +65,6 @@ images="""<img src="gradio_api/file=login.png" alt="login" style="max-width: 35%
 
 with gr.Blocks() as demo:
     
-        # Add title and markdown
     with gr.Row():
         gr.HTML(title)    
     
@@ -78,6 +84,8 @@ with gr.Blocks() as demo:
         gr.HTML(description)
         gr.HTML(images)
 
+    
+    
     homeworks_btn.click(fn=fetch_homework,
                         inputs=[date],
                         outputs=homeworks_output)
@@ -87,4 +95,5 @@ with gr.Blocks() as demo:
                         inputs=[date],
                         outputs=timetable_output)
 
+
 demo.launch(mcp_server=True,allowed_paths=["/"])

scrape_fake_app.py

@@ -1,14 +1,22 @@
 from playwright.sync_api import sync_playwright
 import os
 from dotenv import load_dotenv
-import json
+import time
+import base64
 
-def load_credentials()-> dict:
-    load_dotenv()
-    URL = os.getenv('FAKE_APP_URL')
-    USERNAME = os.getenv('FAKE_APP_USERNAME')
-    PASSWORD = os.getenv('FAKE_APP_PASSWORD')
-    return (URL,USERNAME,PASSWORD)
+# def load_credentials()-> dict:
+#     load_dotenv()
+#     URL = os.getenv('FAKE_APP_URL')
+#     USERNAME = os.getenv('FAKE_APP_USERNAME')
+#     PASSWORD = os.getenv('FAKE_APP_PASSWORD')
+#     return (URL,USERNAME,PASSWORD)
+
+def decode_token(base64_token) -> str:
+    base64_bytes = base64_token.encode("ascii")
+    token_bytes = base64.b64decode(base64_bytes)
+    token = token_bytes.decode("ascii")
+    # print(token)
+    return token
 
 def extract_homework_text(page) -> str:
     card = page.get_by_title("homework")
@@ -45,47 +53,44 @@ def extract_timetable_text(page):
 
 
 # print(URL,USERNAME,PASSWORD)
-def get_homework() -> str:
-    try:
-        URL,USERNAME,PASSWORD=load_credentials()
-        
-        with sync_playwright() as playwright:
-            browser = playwright.firefox.launch(headless=True)
-            page = browser.new_page()
-            page.goto(URL,wait_until="domcontentloaded")
-            
-            page.get_by_role('textbox',name='username').fill(USERNAME)
-            page.get_by_role('textbox',name='password').fill(PASSWORD)
-            page.get_by_role('button',name='login').click()
-            page.wait_for_url("**/dashboard")
-            # page.wait_for_timeout(1000)         
-            homework = extract_homework_text(page)
-            browser.close()        
-        return homework
+def get_homework(base64_token) -> str:
+    token=decode_token(base64_token)
+    # print(token)
+    URL,USERNAME,PASSWORD=token.split("<sep>")
     
-    except Exception as e:
-        return json.dumps(e)
-
-def get_timetable() -> str:
-    try:
-        URL,USERNAME,PASSWORD=load_credentials()
+    with sync_playwright() as playwright:
+        browser = playwright.firefox.launch(headless=True)
+        page = browser.new_page()
+        page.goto(URL,wait_until="domcontentloaded")
         
-        with sync_playwright() as playwright:
-            browser = playwright.firefox.launch(headless=True)
-            page = browser.new_page()
-            page.goto(URL,wait_until="domcontentloaded")
-            
-            page.get_by_role('textbox',name='username').fill(USERNAME)
-            page.get_by_role('textbox',name='password').fill(PASSWORD)
-            page.get_by_role('button',name='login').click()
-            page.wait_for_url("**/dashboard")
-            # page.wait_for_timeout(1000)        
-            timetable = extract_timetable_text(page)
-            browser.close()
-        return timetable
+        page.get_by_role('textbox',name='username').fill(USERNAME)
+        page.get_by_role('textbox',name='password').fill(PASSWORD)
+        page.get_by_role('button',name='login').click()
+        page.wait_for_url("**/dashboard")
+        # page.wait_for_timeout(1000)         
+        homework = extract_homework_text(page)
+        browser.close()        
+    return homework
+
+def get_timetable(base64_token) -> str:
+    token=decode_token(base64_token)
+    URL,USERNAME,PASSWORD=token.split("<sep>")
+    # URL,USERNAME,PASSWORD=load_credentials()
     
-    except Exception as e:
-        return json.dumps(e)
+    with sync_playwright() as playwright:
+        browser = playwright.firefox.launch(headless=True)
+        page = browser.new_page()
+        page.goto(URL,wait_until="domcontentloaded")
+        
+        page.get_by_role('textbox',name='username').fill(USERNAME)
+        page.get_by_role('textbox',name='password').fill(PASSWORD)
+        page.get_by_role('button',name='login').click()
+        page.wait_for_url("**/dashboard")
+        # page.wait_for_timeout(1000)        
+        timetable = extract_timetable_text(page)
+        browser.close()
+        
+    return timetable
 
 if __name__=="__main__":
     print(get_homework())