Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import json
 import os
 import pdfplumber
+import requests
 import together
 from sentence_transformers import SentenceTransformer
 import faiss
@@ -13,6 +14,8 @@ from flask import jsonify
 
 load_dotenv()
 
+API_URL = f"https://b4e0-105-160-26-239.ngrok-free.app/post_embeddings"
+
 # Set up Together.AI API Key (Replace with your actual key)
 assert os.getenv("TOGETHER_API_KEY"), "api key missing"
 
@@ -40,8 +43,17 @@ def store_document_data(PDF_FILE):
 
         # Generate and return embedding
         embedding = embedding_model.encode([text]).astype(np.float32)
+        
         print("Embeddings generated")
-
+        print(f"sending to {API_URL}")
+
+        try:
+            response = requests.post(API_URL, json=payload)
+            response_data = response.json()
+            return response_data
+        except requests.exceptions.RequestException as e:
+            return {"error": str(e)}
+    
         return embedding
     else:
         return "No PDF file provided."