Update app.py

app.py

@@ -5,6 +5,12 @@ import asyncio
 
 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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",
 )
@@ -27,7 +33,26 @@ async def read_root(messages: list[dict]):
         messages=messages,
         model="llama3-70b-8192",
     )
-    return chat_completion.choices[0].message.content
+   return chat_completion.choices[0].message.content
+
+
+@app.post("/upload-image/")
+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, где допущина ошибка, также дай координаты мест где нужна была запятая(квадратом)."]
+    )
+
+    return {"filename": file.filename, "location": file_location}