update code

app.py

@@ -1,12 +1,12 @@
 import asyncio
 from pydantic_ai.result import ResultData, RunResult
 import streamlit as st
-from pydantic_ai import Agent
+from pydantic_ai import Agent,RunContext, Tool
 from pydantic_ai.models.groq import GroqModel
 import nest_asyncio
 import pdfplumber
-from transformers import pipeline
-import torch
+#from transformers import pipeline
+#import torch
 import os
 import presentation as customClass
 from streamlit_pdf_viewer import pdf_viewer
@@ -23,11 +23,12 @@ model = GroqModel('llama-3.1-70b-versatile', api_key = api_key)
 
 
 # to summarize
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+#summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 #summarizer = pipeline('text2text-generation', model='describeai/gemini')
 #nlpaueb/legal-bert-base-uncased
 
 
+
 def split_long_string(long_string, chunk_size=3500):
     string_data =  "".join(long_string)
     words = string_data.split()
@@ -35,10 +36,16 @@ def split_long_string(long_string, chunk_size=3500):
     
     return chunks
 
+def return_data() -> str:
+    return "".join(data)
+
 
 async def ppt_content(data):
     agent = Agent(model,
-    #result_type=customClass.PPT,
+    result_type=customClass.PPT,
+    tools=[
+        Tool(return_data,takes_ctx=False)
+    ],
     system_prompt=(
         "You are an expert in making power-point perssentation",
         "Create 6 sliders",
@@ -65,18 +72,18 @@ async def ppt_content(data):
     
     
     
-    result_1 = agent.run_sync(user_prompt = "".join(data))
+    result_1 = agent.run_sync(user_prompt = "Create a power point presentation with 6 slides")
     st.text(result_1.data)
     print(result_1.data)
     
 
 def ai_ppt(data):
     #call summerizer to summerize pdf 
-    summary = summarizer("".join(data), max_length=400, min_length=100, truncation=True,do_sample=False)
+    # summary = summarizer("".join(data), max_length=400, min_length=100, truncation=True,do_sample=False)
     
-    summary_texts = [item['summary_text'] for item in summary]
+    # summary_texts = [item['summary_text'] for item in summary]
     #summary_texts = [item['generated_text'] for item in summary]
-    asyncio.run(ppt_content(data=summary_texts))
+    asyncio.run(ppt_content(data=data))
 
 
 def extract_data(feed):