Update main.py

main.py

@@ -354,7 +354,46 @@ async def root():
     if html_content is None:
         return HTMLResponse(content="<h1>File not found</h1>", status_code=404)
     return HTMLResponse(content=html_content)
-
+    
+@app.get("/dynamo", response_class=HTMLResponse)
+async def dynamic_ai_page(request: Request):
+    user_agent = request.headers.get('user-agent', 'Unknown User')
+    client_ip = request.client.host
+    location = f"IP: {client_ip}"
+    
+    prompt = f"""
+    Generate a dynamic HTML page for a user with the following details: with name "LOKI.AI" 
+    - User-Agent: {user_agent}
+    - Location: {location}
+    - Style: Cyberpunk, minimalist, or retro
+    
+    Make sure the HTML is clean and includes a heading, also have cool animations a motivational message, and a cool background.
+    Wrap the generated HTML in triple backticks (```).
+    """
+    
+    payload = {
+        "model": "open-mistral-nemo",
+        "messages": [{"role": "user", "content": prompt}]
+    }
+    
+    headers = {
+        "Authorization": "Bearer playground"
+    }
+    
+    response = requests.post("https://parthsadaria-lokiai.hf.space/chat/completions", json=payload, headers=headers)
+    data = response.json()
+    
+    # Extract HTML from ``` blocks
+    html_content = re.search(r"```(.*?)```", data['choices'][0]['message']['content'], re.DOTALL)
+    if html_content:
+        html_content = html_content.group(1).strip()
+    
+    # Remove the first word
+    if html_content:
+        html_content = ' '.join(html_content.split(' ')[1:])
+    
+    return HTMLResponse(content=html_content)
+    
 @app.get("/playground", response_class=HTMLResponse)
 async def playground():
     html_content = read_html_file("playground.html")