Spaces:

Shipmaster1
/

Agent_Workout_531

Runtime error

App Files Files Community

T-K-O-H commited on Apr 22, 2025

Commit

e2239a8

1 Parent(s): d5c37af

Convert to ChatUI with FastAPI backend

Browse files

Files changed (4) hide show

Dockerfile +6 -7
app/main.py +80 -0
frontend/index.html +95 -0
requirements.txt +11 -6

Dockerfile CHANGED Viewed

@@ -8,7 +8,7 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 # Create a non-root user
-RUN useradd -m -u 1000 chainlit_user
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
@@ -18,20 +18,19 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 # Create necessary directories and set permissions
-RUN mkdir -p data/uploads .files .chainlit/translations && \
-    chown -R chainlit_user:chainlit_user /app
 # Set environment variables
 ENV PYTHONPATH=/app
-ENV CHAINLIT_SERVER_PORT=8000
-ENV CHAINLIT_HOST=0.0.0.0
 ENV HOME=/app
 # Switch to non-root user
-USER chainlit_user
 # Expose the port
 EXPOSE 8000
 # Command to run the application
-CMD ["chainlit", "run", "app/chainlit/chainlit.py", "--port", "8000", "--host", "0.0.0.0"]

     && rm -rf /var/lib/apt/lists/*
 # Create a non-root user
+RUN useradd -m -u 1000 app_user
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 COPY . .
 # Create necessary directories and set permissions
+RUN mkdir -p data/uploads .files frontend && \
+    chown -R app_user:app_user /app
 # Set environment variables
 ENV PYTHONPATH=/app
+ENV PORT=8000
 ENV HOME=/app
 # Switch to non-root user
+USER app_user
 # Expose the port
 EXPOSE 8000
 # Command to run the application
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"]

app/main.py ADDED Viewed

	@@ -0,0 +1,80 @@

+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from typing import List, Optional
+import os
+import json
+from app.agent.agent import ResearchAgent
+from app.agent.tools.semantic_search import SemanticSearch
+from app.agent.tools.citation_analyzer import CitationAnalyzer
+from app.agent.tools.workout_planner import WorkoutPlanner
+app = FastAPI()
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+# Initialize tools and agent
+semantic_search = SemanticSearch()
+citation_analyzer = CitationAnalyzer()
+workout_planner = WorkoutPlanner()
+tools = [semantic_search, citation_analyzer, workout_planner]
+agent = ResearchAgent(tools)
+class Message(BaseModel):
+    role: str
+    content: str
+class ChatRequest(BaseModel):
+    messages: List[Message]
+    stream: Optional[bool] = False
+@app.post("/chat")
+async def chat(request: ChatRequest):
+    try:
+        # Get the last message from the user
+        last_message = request.messages[-1]
+        if last_message.role != "user":
+            raise HTTPException(status_code=400, detail="Last message must be from user")
+        # Process the message with the agent
+        response = agent.process_message(last_message.content)
+        return JSONResponse(content={
+            "choices": [{
+                "message": {
+                    "role": "assistant",
+                    "content": response
+                }
+            }]
+        })
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+@app.post("/upload")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        # Save the file
+        file_path = os.path.join("data/uploads", file.filename)
+        with open(file_path, "wb") as buffer:
+            content = await file.read()
+            buffer.write(content)
+        return JSONResponse(content={
+            "message": "File uploaded successfully",
+            "filename": file.filename
+        })
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)

frontend/index.html ADDED Viewed

	@@ -0,0 +1,95 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>5/3/1 Workout Program Assistant</title>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@huggingface/chat-ui/dist/chat-ui.css">
+    <style>
+        body {
+            margin: 0;
+            padding: 0;
+            font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, 'Open Sans', 'Helvetica Neue', sans-serif;
+        }
+        #chat-container {
+            max-width: 800px;
+            margin: 0 auto;
+            height: 100vh;
+            display: flex;
+            flex-direction: column;
+        }
+        .header {
+            padding: 1rem;
+            background-color: #f5f5f5;
+            border-bottom: 1px solid #e0e0e0;
+        }
+        .header h1 {
+            margin: 0;
+            font-size: 1.5rem;
+            color: #333;
+        }
+    </style>
+</head>
+<body>
+    <div id="chat-container">
+        <div class="header">
+            <h1>5/3/1 Workout Program Assistant</h1>
+        </div>
+        <div id="chat"></div>
+    </div>
+    <script src="https://cdn.jsdelivr.net/npm/@huggingface/chat-ui/dist/chat-ui.js"></script>
+    <script>
+        const chat = new ChatUI({
+            container: document.getElementById('chat'),
+            apiEndpoint: '/chat',
+            uploadEndpoint: '/upload',
+            welcomeMessage: 'Welcome to the 5/3/1 Workout Program Assistant! How can I help you today?',
+            userMessageColor: '#4a90e2',
+            assistantMessageColor: '#f5f5f5',
+            inputPlaceholder: 'Type your message here...',
+            uploadButtonText: 'Upload PDF',
+            uploadAcceptedFileTypes: '.pdf',
+            onMessage: async (message) => {
+                try {
+                    const response = await fetch('/chat', {
+                        method: 'POST',
+                        headers: {
+                            'Content-Type': 'application/json',
+                        },
+                        body: JSON.stringify({
+                            messages: [{
+                                role: 'user',
+                                content: message
+                            }]
+                        })
+                    });
+                    const data = await response.json();
+                    return data.choices[0].message.content;
+                } catch (error) {
+                    console.error('Error:', error);
+                    return 'Sorry, there was an error processing your request.';
+                }
+            },
+            onFileUpload: async (file) => {
+                try {
+                    const formData = new FormData();
+                    formData.append('file', file);
+                    const response = await fetch('/upload', {
+                        method: 'POST',
+                        body: formData
+                    });
+                    const data = await response.json();
+                    return `File "${data.filename}" uploaded successfully. How can I help you with this file?`;
+                } catch (error) {
+                    console.error('Error:', error);
+                    return 'Sorry, there was an error uploading your file.';
+                }
+            }
+        });
+    </script>
+</body>
+</html>

requirements.txt CHANGED Viewed

@@ -1,6 +1,11 @@
-langchain>=0.1.0
-langchain-openai>=0.0.5
-chainlit>=0.7.0
-PyPDF2>=3.0.0
-python-dotenv>=1.0.0
-huggingface-hub>=0.16.0

+fastapi==0.104.1
+uvicorn==0.24.0
+python-multipart==0.0.6
+pydantic==2.4.2
+openai==1.3.0
+python-dotenv==1.0.0
+langchain==0.0.350
+pypdf==3.17.1
+numpy==1.24.3
+pandas==2.0.3
+scikit-learn==1.3.2