hostserver3

Runtime error

App Files Files Community

abdullahalioo commited on May 22

Commit

85582c9

verified ·

1 Parent(s): 2f3bde3

Update main.py

Browse files

Files changed (1) hide show

main.py +44 -76

main.py CHANGED Viewed

@@ -1,8 +1,5 @@
-import os
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-from hugchat import hugchat
-from hugchat.login import Login
 import logging
 # Set up logging
@@ -12,78 +9,49 @@ logger = logging.getLogger(__name__)
 # Initialize FastAPI app
 app = FastAPI()
-# HuggingFace credentials from environment variables
-EMAIL = "abdullahali1692011@gmail.com"
-PASSWORD = "Allahisgreatest17"
-ASSISTANT_ID = "682e0c1f5f0c3d952a27498e"
-# Global chatbot instance
-chatbot = None
-# Pydantic model for request body
-class PromptRequest(BaseModel):
-    prompt: str
-# Lifespan handler for startup and shutdown
-from contextlib import asynccontextmanager
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    global chatbot
-    try:
-        # Validate credentials
-        if not EMAIL or not PASSWORD or not ASSISTANT_ID:
-            logger.error("Missing environment variables: EMAIL=%s, PASSWORD=%s, ASSISTANT_ID=%s",
-                         EMAIL, "****" if PASSWORD else None, ASSISTANT_ID)
-            raise ValueError("HuggingFace credentials or assistant ID not provided in environment variables")
-        # Log in to HuggingFace
-        sign = Login(EMAIL, PASSWORD)
-        try:
-            # Try login without cookie_path_dir to avoid file system issues
-            cookies = sign.login(save_cookies=False)
-            logger.info("Successfully logged in to HuggingFace without saving cookies")
-        except Exception as e:
-            logger.error("Login failed: %s", str(e))
-            raise
-        # Initialize chatbot
-        chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
-        chatbot.new_conversation(assistant=ASSISTANT_ID, switch_to=True)
-        logger.info("Chatbot initialized with assistant ID: %s", ASSISTANT_ID)
-    except Exception as e:
-        logger.error("Failed to initialize chatbot: %s", str(e))
-        raise HTTPException(status_code=500, detail=f"Failed to initialize chatbot: {str(e)}")
-    yield  # Application runs here
-    # Shutdown: Clean up (optional)
-    logger.info("Shutting down application")
-# Attach lifespan handler to FastAPI app
-app.lifespan = lifespan
-# Endpoint to handle chat requests
-@app.post("/ask")
-async def ask(prompt_request: PromptRequest):
-    try:
-        if not chatbot:
-            logger.error("Chatbot not initialized")
-            raise HTTPException(status_code=500, detail="Chatbot not initialized")
-        # Send prompt to chatbot
-        response = chatbot.chat(prompt_request.prompt, stream=False)
-        logger.info("Received response for prompt: %s", prompt_request.prompt)
-        # Return structured response
-        return {"status": "success", "response": response}
-    except Exception as e:
-        logger.error("Error processing request: %s", str(e))
-        raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
 # Health check endpoint
 @app.get("/health")
 async def health():
-    return {"status": "healthy", "chatbot_initialized": chatbot is not None}

+from fastapi import FastAPI
+from fastapi.responses import HTMLResponse
 import logging
 # Set up logging
 # Initialize FastAPI app
 app = FastAPI()
+# Serve HTML page
+@app.get("/", response_class=HTMLResponse)
+async def serve_html():
+    logger.info("Serving HTML page")
+    return """
+    <html>
+    <head>
+        <title>AI Chemistry Chat</title>
+        <style>
+            body { font-family: Arial, sans-serif; margin: 20px; }
+            #response { white-space: pre-wrap; border: 1px solid #ccc; padding: 10px; min-height: 100px; }
+            input { width: 300px; padding: 5px; }
+            button { padding: 5px 10px; }
+        </style>
+    </head>
+    <body>
+        <h1>AI Chemistry Chat</h1>
+        <input type="text" id="prompt" placeholder="Ask about chemistry..." value="Tell me something I might not know about chemistry." />
+        <button onclick="sendPrompt()">Send</button>
+        <div id="response"></div>
+        <script src="https://js.puter.com/v2/"></script>
+        <script>
+            async function sendPrompt() {
+                const prompt = document.getElementById('prompt').value;
+                const responseDiv = document.getElementById('response');
+                responseDiv.innerHTML = 'Loading...';
+                try {
+                    const resp = await puter.ai.chat(prompt, { model: 'grok-beta', stream: true });
+                    responseDiv.innerHTML = '';
+                    for await (const part of resp) {
+                        responseDiv.innerHTML += part?.text?.replaceAll('\\n', '<br>') || '';
+                    }
+                } catch (error) {
+                    responseDiv.innerHTML = 'Error: ' + error.message;
+                }
+            }
+        </script>
+    </body>
+    </html>
+    """
 # Health check endpoint
 @app.get("/health")
 async def health():
+    logger.info("Health check requested")
+    return {"status": "healthy"}