Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
-from groq import Groq
+import pdfplumber
+import gradio as gr
 from dotenv import load_dotenv
+from groq import Groq
 
 # Load environment variables from a .env file
 load_dotenv()
@@ -9,13 +11,9 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 # Instantiate the Groq client
 client = Groq(api_key=GROQ_API_KEY)
 
-
-# Load Groq's generative model (the model name 'groq-pro' is illustrative)
-model = groq.GenerativeModel('groq-pro')
-
 def extract_text_from_pdf(pdf_file):
     text = ""
-    with pdfplumber.open(pdf_file) as pdf:
+    with pdfplumber.open(pdf_file.name) as pdf:
         for page in pdf.pages:
             page_text = page.extract_text()
             if page_text:
@@ -33,8 +31,16 @@ def summarize_pdf(pdf_file):
     prompt = f"Summarize the following PDF content:\n\n{text}"
 
     try:
-        response = model.generate_content(prompt)  # Adjust parameters per Groq's API
-        return response.text.strip()
+        response = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ],
+            model="llama3-8b-8192",  # Replace with your desired model ID
+        )
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"Error during summarization: {e}"