Update api/main.py

api/main.py

@@ -1,24 +1,18 @@
-#main.py
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import google.generativeai as genai
 
-# Cấu hình FastAPI
 app = FastAPI()
 
-# Cấu hình API key cho Gemini
 genai.configure(api_key='AIzaSyBvhkco5aRtkqEyPFFaTKaaDmJkgU12MJU')
 llm_model = genai.GenerativeModel('gemini-1.5-pro')
 
-# Khai báo request body để nhận nội dung từ Streamlit
 class InputText(BaseModel):
     text: str
 
-# Route xử lý yêu cầu từ Streamlit
 @app.post("/generate")
 async def generate_text(input_text: InputText):
     try:
-        # Gọi API Gemini
         response = llm_model.generate_content(input_text.text)
         return {"generated_text": response.text}
     except Exception as e: