first push

README.md

@@ -1,8 +1,8 @@
 ---
-title: App
-emoji: ⚡
-colorFrom: gray
-colorTo: pink
+title: Streamlit
+emoji: 🏃
+colorFrom: yellow
+colorTo: red
 sdk: streamlit
 sdk_version: 1.32.2
 app_file: app.py

SM2.py

@@ -0,0 +1,37 @@
+import requests
+import streamlit as st
+import json
+
+
+model = st.selectbox('which model would you like to use',
+                    ('MobileBERT', 'BERT', 'DeBERTa-v2'))
+
+
+userInput = st.text_input(f"write to the {model}")
+
+contextSelect = st.radio("Pick a context mode:", ["File", "Text"])
+if contextSelect == "File":
+    userContext = st.file_uploader("Pick a file for the context", accept_multiple_files=True)
+    context = "/uploadfile"
+else:
+    userContext = st.text_input("write the context")
+    context = "/contextText"
+
+siteUrl = "http://127.0.0.1:8000"
+
+if st.button("Envoyer la requête"):
+    params = {'texte': userInput, "model": model}
+    if userContext is not None:
+        if contextSelect == "File":
+            files = {"file": (userContext.name, userContext, userContext.type)}
+            response = requests.post(
+                siteUrl+context, params=params, files=files)
+        else:
+            params["context"] = userContext
+            print(params)
+            response = requests.post(siteUrl+context, params=params)
+    else:
+        response = requests.post(siteUrl+'/withoutFile', params=params)
+
+    st.write("Statut de la requête:", response.status_code)
+    st.write("Réponse du serveur:", response.text)

app.py

@@ -0,0 +1,51 @@
+import streamlit as st
+import requests
+import streamlit as st
+
+siteUrl = "http://127.0.0.1:8000"
+
+userContext = None
+context = None
+
+with st.sidebar:
+    model = st.selectbox('which model would you like to use',
+                         ('MobileBERT', 'BERT', 'DeBERTa-v2'))
+    contextSelect = st.radio("Pick a context mode:", ["File", "Text"])
+    if contextSelect == "File":
+        userContext = st.file_uploader("Pick a file for the context", accept_multiple_files=True)
+        context = "/uploadfile"
+    else:
+        userContext = st.text_input("write the context")
+        context = "/contextText"
+
+st.title("ALOQAS")
+
+"""
+bla bla bla
+"""
+
+if "messages" not in st.session_state:
+    st.session_state["messages"] = [
+        {"role": "assistant", "content": "Hi, I'm ALOQAS. How can I help you?"}
+    ]
+
+for msg in st.session_state.messages:
+    st.chat_message(msg["role"]).write(msg["content"])
+
+if prompt := st.chat_input(placeholder="Write to ALOQAS..."):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    st.chat_message("user").write(prompt)
+    params = {'texte': prompt, "model": model}
+    if userContext is not None:
+        if contextSelect == "File" and userContext:
+            # Préparation de la liste des fichiers pour l'envoi
+            files = [("file", (file.name, file, file.type)) for file in userContext]
+            response = requests.post(siteUrl + context, params=params, files=files)
+        else:
+            params["context"] = userContext
+            response = requests.post(siteUrl + context, params=params)
+    else:
+        response = requests.post(siteUrl + '/withoutFile', params=params)
+
+    st.write("Statut de la requête:", response.status_code)
+    st.write("Réponse du serveur:", response.text)

main.py

@@ -0,0 +1,63 @@
+from fastapi import FastAPI, HTTPException, File, UploadFile
+from fastapi.middleware.cors import CORSMiddleware
+from typing import Annotated
+import uvicorn
+from fastapi import FastAPI, UploadFile, File 
+from typing import Union 
+import json 
+import csv
+
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+@app.on_event("startup")
+async def startup_event():
+    print("start")
+
+
+@app.get("/")
+async def root():
+    return {"message": "Hello World"}
+
+
+@app.post("/uploadfile/")
+async def create_upload_file(file: UploadFile, texte: str, model: str):
+
+    return {"model": model, "texte": texte, "filename": file.filename}
+
+
+@app.post("/contextText/")
+async def create_upload_file(context: str, texte: str, model: str):
+
+    return {"model": model, "texte": texte, "context": context}
+
+
+@app.post("/withoutFile/")
+async def create_upload_file(texte: str, model: str):
+
+    return {"model": model, "texte": texte}
+
+
+def extract_data(file: UploadFile) -> Union[str, dict, list]:
+    if file.filename.endswith(".txt"):
+        data = file.file.read()
+        return data.decode("utf-8")
+    elif file.filename.endswith(".csv"):
+        data = file.file.read().decode("utf-8")
+        rows = data.split("\n")
+        reader = csv.DictReader(rows)
+        return [dict(row) for row in reader]
+    elif file.filename.endswith(".json"):
+        data = file.file.read().decode("utf-8")
+        return json.loads(data)
+    else:
+        return "Invalid file format"

require.txt

@@ -0,0 +1,4 @@
+pip3 install python-multipart
+pip3 install fastapi
+pip3 install uvicorn
+pip3 install fastapi.middleware.cors

requirements.txt

@@ -0,0 +1,6 @@
+transformers
+torch
+python-multipart
+fastapi
+uvicorn
+fastapi.middleware.cors