Upload 2 files

app.py

@@ -0,0 +1,22 @@
+# app.py
+import streamlit as st
+import requests
+
+st.title("Gemini API Tester")
+
+input_text = st.text_area("Type something...")
+
+if st.button("Send"):
+    try:
+        response = requests.post(
+            "http://fastapi:8000/generate", 
+            json={"text": input_text}
+        )
+        response_data = response.json()
+
+        if response.status_code == 200:
+            st.success(response_data["generated_text"])
+        else:
+            st.error(f"Error: {response_data.get('detail', 'Unknown error')}")
+    except Exception as e:
+        st.error(f"Request failed: {e}")

main.py

@@ -0,0 +1,25 @@
+#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:
+        raise HTTPException(status_code=500, detail=f"Error generating content: {e}")