Update app.py

app.py

@@ -1,117 +1,79 @@
-# 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
+# Use the official Python 3.9 image as the base image
+FROM python:3.9
 
+# 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
 
-# model = SentenceTransformer('flax-sentence-embeddings/all_datasets_v4_MiniLM-L6')
-model = SentenceTransformer("sentence-transformers/all-roberta-large-v1")
+# Set the PYNGROK_CONFIG environment variable
+ENV PYNGROK_CONFIG /tmp/pyngrok.yml
 
+# Set the NGROK_PATH environment variable to a writable location
+ENV NGROK_PATH /tmp/ngrok
 
+# Copy requirements.txt into the container
+COPY requirements.txt .
 
+# 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=['*'],
-)
-
-
-# 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)
-    
-    
-    dict={ }
-
-    c=0
-    text_e= text_e.tolist()
-
-    for num in text_e:
-        dict[c]= num
-        c= c+1
-
-    
-    
-    return dict
-
-    
-    
-   
-   
-    return "hello"
-    
+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
 
+# 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"]