Update main.py

main.py

@@ -25,9 +25,10 @@ load_dotenv()
 
 # Configure Gemini API
 secret = os.environ["GEMINI"]
+SECRET_KEY = os.environ["SECRET_KEY"]
 genai.configure(api_key=secret)
 model_vision = genai.GenerativeModel('gemini-1.5-flash')
-model_text = genai.GenerativeModel('gemini-pro')
+model_text = genai.GenerativeModel('gemini-1.5-flash')
 
 
 
@@ -78,7 +79,13 @@ def vision(file_content):
 
 
 @app.post("/get_ocr_data/")
-async def get_data(input_file: UploadFile = File(...)):
+async def get_data(input_file: UploadFile = File(...),user_id:int,token:str):
+        # To decode and verify the token
+        try:
+            decoded_payload = jwt.decode(token, SECRET_KEY, algorithms=["HS256"])
+            print(f"Decoded payload: {decoded_payload}")
+        except:
+            return "Invalid token"
     #try:
         # Determine the file type by reading the first few bytes
         file_content = await input_file.read()
@@ -117,7 +124,7 @@ async def get_data(input_file: UploadFile = File(...)):
         response = model_text.generate_content(prompt)
         print(response.text)
         data = json.loads(response.text.replace("JSON", "").replace("json", "").replace("```", ""))
-        return {"data": data}
+        return {"data": data,"user_id":user_id}
 
     #except Exception as e:
         #raise HTTPException(status_code=500, detail=f"Error processing file: {str(e)}")