Spaces:

microhum
/

Nurse_LLM_Demo

Sleeping

microhum commited on Nov 16, 2024

Commit

b4525ce

1 Parent(s): 982f969

Init: huggingface space rest api

Files changed (4) hide show

DockerFile ADDED Viewed

+FROM python:3.10.9
+COPY . .
+WORKDIR /
+RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md CHANGED Viewed

@@ -1,5 +1,7 @@
 # Nurse LLM
 ```
 /your_path/MALI_Nurse/
 ├── cli.py
@@ -34,9 +36,9 @@
    Alternatively, you can use Poetry for dependency management:
-    ```sh
-    poetry install
-    ```
 3. Set up your environment variables in a `.env` file:
    ```env
@@ -100,18 +102,16 @@ Once the CLI is running, you can use the following commands:
 - To start interacting with the nurse LLM:
-    ```sh
-    start
-    ```
 - To display help information:
-    ```sh
-    help
-    ```
 - To exit the CLI session:
-    ```sh
-    exit
-    ```

 # Nurse LLM
+> #### M.A.R.I.N: Medical Assistance for Record Inference and Nursing Translation
 ```
 /your_path/MALI_Nurse/
 ├── cli.py
    Alternatively, you can use Poetry for dependency management:
+   ```sh
+   poetry install
+   ```
 3. Set up your environment variables in a `.env` file:
    ```env
 - To start interacting with the nurse LLM:
+  ```sh
+  start
+  ```
 - To display help information:
+  ```sh
+  help
+  ```
 - To exit the CLI session:
+  ```sh
+  exit
+  ```

llm/__pycache__/client.cpython-311.pyc CHANGED Viewed

Binary files a/llm/__pycache__/client.cpython-311.pyc and b/llm/__pycache__/client.cpython-311.pyc differ

main.py CHANGED Viewed

@@ -1,6 +1,8 @@
 import uvicorn
 from llm.llm import VirtualNurseLLM
 from fastapi import FastAPI
 from pydantic import BaseModel
 import os
 import dotenv
@@ -22,9 +24,33 @@ nurse_llm = VirtualNurseLLM(
 app = FastAPI()
 class UserInput(BaseModel):
     user_input: str
 @app.get("/history")
 def get_chat_history():
     return {"chat_history": nurse_llm.chat_history}

 import uvicorn
 from llm.llm import VirtualNurseLLM
 from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 import os
 import dotenv
 app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 class UserInput(BaseModel):
     user_input: str
+@app.get("/", response_class=HTMLResponse)
+def read_index():
+    return """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>MALI_NURSE API/title>
+    </head>
+    <body>
+        <h1>Welcome to MALI_NURSE API</h1>
+        <p>This is the index page. Use the link below to access the API docs:</p>
+        <a href="/docs">Go to Swagger Docs UI</a>
+    </body>
+    </html>
+    """
 @app.get("/history")
 def get_chat_history():
     return {"chat_history": nurse_llm.chat_history}