Update app.py

app.py

@@ -3,61 +3,28 @@ from typing import Union
 from fastapi import FastAPI
 import asyncio
 
-from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import HTMLResponse
-import shutil
-import os
 from groq import Groq, AsyncGroq
 
-import google.generativeai as genai
-import os
-
-genai.configure(api_key="AIzaSyBGhEOy-JYMzGtTcRjBjP51OGR168WKRFw")
-
-
 client = AsyncGroq(
-    api_key="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",
+    api_key="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",
 )
 
 SYSTEM_PROMPT = """
-You are helpful assistant, your task is to help me operating my system!
+Ты ассистент, помогай людям!
 """
 
 app = FastAPI()
 
 
-# @app.post("/get_response")
-# async def read_root(messages: list[dict]):
-#     messages.insert(0, {
-#             "role": "system",
-#             "content": SYSTEM_PROMPT
-#         }
-#     )
-#     chat_completion = await client.chat.completions.create(
-#         messages=messages,
-#         model="llama3-70b-8192",
-#     )
-#    return chat_completion.choices[0].message.content
-
-
-@app.get("/test")
-def test():
-    return "Hello"
-
-@app.post("/")
-async def upload_image(file: UploadFile = File(...)):
-    os.makedirs("uploads", exist_ok=True)
-
-    file_location = f"uploads/{file.filename}"
-    with open(file_location, "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
-
-    myfile = genai.upload_file(file_location)
-
-    model = genai.GenerativeModel("gemini-1.5-flash")
-    result = model.generate_content(
-        [myfile, "\n\n",
-         "Проверь работу на фото, дай координаты всех слов в формате OpenCV, где допущина ошибка, также дай координаты мест где нужна была запятая(квадратом)."]
+@app.post("/get_response")
+async def read_root(messages: list[dict]):
+    messages.insert(0, {
+            "role": "system",
+            "content": SYSTEM_PROMPT
+        }
     )
-
-    return result.text
+    chat_completion = await client.chat.completions.create(
+        messages=messages,
+        model="llama3-70b-8192",
+    )
+    return chat_completion.choices[0].message.content