Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr 
 import os
 from google import genai
+from google.genai import types
 from PyPDF2 import PdfReader
 from docx import Document
 
@@ -99,31 +100,17 @@ def extract_text_from_docx(docx_file):
         text += para.text + "\n"
     return text
 
-def generate_response(message: str, system_prompt: str, temperature: float, max_tokens: int):
-    # conversation = [
-    #     {"role": "system", "content": system_prompt},
-    #     {"role": "user", "content": message}
-    # ]
-
-    # response = client.chat.completions.create(
-    #     model="llama-3.1-8B-Instant",
-    #     messages=conversation,
-    #     temperature=temperature,
-    #     max_tokens=max_tokens,
-    #     stream=False
-    # )
-
+def generate_response(message: str, system_prompt: str, temperature: float, max_tokens: int):  
     response = client.models.generate_content(
         model="gemini-2.5-flash",
         contents=message,
-        config=GenerateContentConfig(
+        config=types.GenerateContentConfig(
             system_instruction=system_prompt,
             temperature=temperature,
             max_output_tokens=max_tokens
         )
     )
 
-    # return response.choices[0].message.content
     return response.text