Update app.py

app.py

@@ -1,79 +1,121 @@
-# Use the official Python 3.9 image as the base image
-FROM python:3.9
+# import firebase_admin
+# from firebase_admin import credentials
+# from firebase_admin import firestore
+import io
+from fastapi import FastAPI, File, UploadFile
+# from werkzeug.utils import secure_filename
+# import speech_recognition as sr
+import subprocess
+import os
+import requests
+import random
+import pandas as pd
+# from pydub import AudioSegment
+from datetime import datetime
+from datetime import date
+# import numpy as np
+# from sklearn.ensemble import RandomForestRegressor
+# import shutil
+import json
+# from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
+from pydantic import BaseModel
+from typing import Annotated
+# from transformers import BertTokenizerFast, EncoderDecoderModel
+import torch
+import re
+# from transformers import AutoTokenizer, T5ForConditionalGeneration
+from fastapi import Form
+# from transformers import AutoModelForSequenceClassification
+# from transformers import TFAutoModelForSequenceClassification
+# from transformers import AutoTokenizer, AutoConfig
+# import numpy as np
+# from scipy.special import softmax
+from sentence_transformers import SentenceTransformer
 
-# Expose the port
-EXPOSE 7860
 
-# Keeps Python from generating .pyc files in the container
-ENV PYTHONDONTWRITEBYTECODE=1
 
-# Turns off buffering for easier container logging
-ENV PYTHONUNBUFFERED=1
 
-# Set the PYNGROK_CONFIG environment variable
-ENV PYNGROK_CONFIG /tmp/pyngrok.yml
+# model = SentenceTransformer('flax-sentence-embeddings/all_datasets_v4_MiniLM-L6')
+# model = SentenceTransformer("sentence-transformers/all-roberta-large-v1")
+# model =SentenceTransformer("intfloat/multilingual-e5-large")
 
-# Set the NGROK_PATH environment variable to a writable location
-ENV NGROK_PATH /tmp/ngrok
 
-# Copy requirements.txt into the container
-COPY requirements.txt .
+model = SentenceTransformer('intfloat/multilingual-e5-large')
 
-# Upgrade pip and install the required packages
-RUN pip install --upgrade pip && \
-    pip install -r requirements.txt
 
-# Install sudo and create the necessary directories before copying the files
-RUN apt-get update && \
-    apt-get install -y sudo && \
-    mkdir -p /code/image
+class Query(BaseModel):
+    text: str
 
-# Creates a non-root user with an explicit UID and adds permission to access the /code folder
-RUN adduser -u 5678 --disabled-password --gecos "" appuser && \
-    usermod -aG sudo appuser && \
-    usermod -aG root appuser && \
-    chown -R appuser:appuser /code
+   
 
-# Create the pyngrok bin directory and set the ownership and permissions for appuser
-RUN mkdir -p /usr/local/lib/python3.9/site-packages/pyngrok/bin && \
-    chown -R appuser:appuser /usr/local/lib/python3.9/site-packages/pyngrok/bin && \
-    chmod -R 777 /usr/local/lib/python3.9/site-packages/pyngrok/bin
 
-RUN mkdir -p /.ngrok2 && \
-    chown -R appuser:appuser /.ngrok2 && \
-    chmod -R 777 /.ngrok2
+from fastapi import FastAPI, Request, Depends, UploadFile, File
+from fastapi.exceptions import HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
 
-RUN apt-get update && \
-    apt-get install -y curl
 
-# Set the working directory and copy the files
-WORKDIR /code
+# now = datetime.now()
 
-# Set the ownership and permissions for the /code directory and its contents
-RUN chown -R appuser:appuser /code && \
-    chmod -R 777 /code
 
-COPY . /code
+# UPLOAD_FOLDER = '/files'
+# ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png',
+#                       'jpg', 'jpeg', 'gif', 'ogg', 'mp3', 'wav'}
 
-# RUN chown -R appuser:appuser /code/data.csv && \
-#     chmod -R 777 /code/data.csv
 
-# Copy the pyngrok.yml configuration file
-COPY pyngrok.yml /tmp/pyngrok.yml
+app = FastAPI()
 
-# Set the TRANSFORMERS_CACHE environment variable to a cache directory inside /tmp
-ENV TRANSFORMERS_CACHE /tmp/transformers_cache
-ENV TORCH_HOME /tmp/torch_cache
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=['*'],
+    allow_credentials=True,
+    allow_methods=['*'],
+    allow_headers=['*'],
+)
 
-USER appuser
 
-# Start the application using pyngrok
-# CMD python main.py
-# Get the public IP address and display it
-RUN curl -s https://api.ipify.org | xargs echo "Public IP:"
-RUN pip install gunicorn
+# cred = credentials.Certificate('key.json')
+# app1 = firebase_admin.initialize_app(cred)
+# db = firestore.client()
+# data_frame = pd.read_csv('data.csv')
+
+
+
+@app.on_event("startup")
+async def startup_event():
+   print("on startup")
+
+   
+
+
+@app.post("/")
+async def get_answer(q: Query ):
+    
+    text = q.text
+    # text_e = model.encode(text)
+    input_texts = [text]
+    embeddings = model.encode(input_texts)
+    text_e = embeddings[0]
+    
+    
+    dict={ }
+
+    c=0
+    text_e= text_e.tolist()
+
+    for num in text_e:
+        dict[c]= num
+        c= c+1
+
+    
+    
+    return dict
+
+    
+    
+   
+   
+    return "hello"
+    
+
 
-# Start the Uvicorn server
-# ENTRYPOINT ["python", "main.py"]
-# CMD ["sh", "-c", "python main.py & sleep infinity"]
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860","--workers","2"]