Update app.py

app.py

@@ -8,13 +8,14 @@ from langchain_core.prompts import PromptTemplate
 import chromadb
 import os
 from uuid import uuid4
+from langchain_google_genai import ChatGoogleGenerativeAI
 
 # Initialize persistent client
 CHROMA_PATH = "./vectordb-chroma/gformdb"
 os.makedirs(CHROMA_PATH, exist_ok=True)
 default_user_aspek = "Jika Jawaban salah berikan nilai 0, jika jawaban benar namun tidak tepat berikan nilai 50, jika jawaban benar dan lengkap serta penjelasan baik, beri nilai 100"
 
-def process_files(csv_file, openai_key, user_aspek=default_user_aspek, pdf_file=False, model_name="gpt-4o-mini"):
+def process_files(csv_file, gemini_key, user_aspek=default_user_aspek, pdf_file=False, model):
     try:
         context = ""
 
@@ -57,10 +58,9 @@ def process_files(csv_file, openai_key, user_aspek=default_user_aspek, pdf_file=
             return qa_efficient, Question
 
         # Set OpenAI API
-        if openai_key == "":
-            openai_key = os.getenv('OPENAI_API_KEY')
+        if gemini_key == "":
+            gemini_key = os.getenv('GEMINI_API_KEY')
             
-        os.environ["OPENAI_API_KEY"] = openai_key
         
         # Process CSV (Gradio provides the file path directly)
 
@@ -129,7 +129,7 @@ def process_files(csv_file, openai_key, user_aspek=default_user_aspek, pdf_file=
         )
         
         # Initialize ChatGPT
-        chat = ChatOpenAI(model=model_name)
+        chat = ChatGoogleGenerativeAI(model=model, api_key=gemini_key)
         chain = PROMPT | chat
         
         # Generate evaluation
@@ -195,13 +195,13 @@ def create_interface():
 
             model_name = gr.Textbox(
                 label="openAI model",
-                placeholder="input your openAI model",
-                value="gpt-4o-mini"
+                placeholder="input your Gemini model",
+                value="gemini-1.5-pro"
             )
             
             api_key = gr.Textbox(
                 label="OpenAI API Key",
-                placeholder="Enter your OpenAI API key, default is my API key, use it wisely XD",
+                placeholder="Enter your Gemini API key, default is my API key, use it wisely XD",
                 type="password"
             )