Spaces:

beeguy
/

website-RAG

Sleeping

App Files Files Community

beeguy commited on Dec 9, 2024

Commit

3f7b6a8

1 Parent(s): 7a26dd9

embeddings upload

Browse files

Files changed (3) hide show

app.py +8 -1
static/index.html +5 -2
static/script.js +32 -0

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from fastapi import FastAPI
 from os import getenv
 from langchain_huggingface import HuggingFaceEmbeddings
 from fastapi.responses import HTMLResponse
@@ -11,6 +11,13 @@ embeddings = HuggingFaceEmbeddings(model_name="jinaai/jina-embeddings-v2-small-e
 app.mount("/static", StaticFiles(directory="static"), name="static")
 @app.get("/embeddings")
 def get_embeddings(input: str):
     result = embeddings.embed_query(input)

+from fastapi import FastAPI, Request
 from os import getenv
 from langchain_huggingface import HuggingFaceEmbeddings
 from fastapi.responses import HTMLResponse
 app.mount("/static", StaticFiles(directory="static"), name="static")
+@app.post("/receive-embeddings")
+async def receive_embeddings(request: Request):
+    data = await request.json()
+    embeddings = data.get("embeddings")
+    # Process the embeddings as needed
+    return {"status": "OK"}
 @app.get("/embeddings")
 def get_embeddings(input: str):
     result = embeddings.embed_query(input)

static/index.html CHANGED Viewed

@@ -5,8 +5,8 @@
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Fast API 🤗 Space served with Uvicorn</title>
-    <link rel="stylesheet" href="style.css" />
-    <script type="module" src="script.js"></script>
 </head>
 <body>
@@ -20,11 +20,14 @@
             </p>
             <p>Here you can create embeddings from the text you provide. This text should be submitted to the
                 /embeddings endpoint.</p>
             <form class="text-gen-form">
                 <label for="text-gen-input">Text prompt</label>
                 <input id="text-gen-input" type="text" value="English: Translate There are many ducks. German:" />
                 <button id="text-gen-submit">Submit</button>
                 <button id="download-embeddings" type="button">Download Embeddings</button>
                 <p class="text-gen-output"></p>
             </form>
         </section>

     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Fast API 🤗 Space served with Uvicorn</title>
+    <link rel="stylesheet" href="/static/style.css" />
+    <script type="module" src="/static/script.js"></script>
 </head>
 <body>
             </p>
             <p>Here you can create embeddings from the text you provide. This text should be submitted to the
                 /embeddings endpoint.</p>
+            <p>If you have embeddings, you can upload them instead.</p>
             <form class="text-gen-form">
                 <label for="text-gen-input">Text prompt</label>
                 <input id="text-gen-input" type="text" value="English: Translate There are many ducks. German:" />
                 <button id="text-gen-submit">Submit</button>
                 <button id="download-embeddings" type="button">Download Embeddings</button>
+                <button id="upload-embeddings" type="button">Upload Embeddings</button>
+                <input type="file" id="file-input" style="display: none;" />
                 <p class="text-gen-output"></p>
             </form>
         </section>

static/script.js CHANGED Viewed

@@ -25,6 +25,8 @@ textGenForm.addEventListener('submit', async (event) => {
 });
 const downloadButton = document.getElementById('download-embeddings');
 const updateDownloadButtonState = () => {
     downloadButton.disabled = embeddingsList.length === 0;
@@ -40,7 +42,37 @@ const downloadEmbeddings = () => {
     downloadAnchorNode.remove();
 };
 downloadButton.addEventListener('click', downloadEmbeddings);
 // Initialize button state
 updateDownloadButtonState();

 });
 const downloadButton = document.getElementById('download-embeddings');
+const uploadButton = document.getElementById('upload-embeddings');
+const fileInput = document.getElementById('file-input');
 const updateDownloadButtonState = () => {
     downloadButton.disabled = embeddingsList.length === 0;
     downloadAnchorNode.remove();
 };
+const uploadEmbeddings = () => {
+    fileInput.click();
+};
+fileInput.addEventListener('change', async (event) => {
+    const file = event.target.files[0];
+    if (file) {
+        const reader = new FileReader();
+        reader.onload = async (e) => {
+            try {
+                const embeddings = JSON.parse(e.target.result);
+                embeddingsList = embeddings; // Store uploaded embeddings in the variable
+                updateDownloadButtonState(); // Update button state
+                // Optionally, you can send the embeddings to the server
+                await fetch('/receive-embeddings', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({ embeddings: embeddingsList })
+                });
+            } catch (err) {
+                console.error('Error reading or parsing the file', err);
+            }
+        };
+        reader.readAsText(file);
+    }
+});
 downloadButton.addEventListener('click', downloadEmbeddings);
+uploadButton.addEventListener('click', uploadEmbeddings);
 // Initialize button state
 updateDownloadButtonState();