[FIX] fixes max_len and min_len dynamic.

tool.py

@@ -17,7 +17,7 @@ class TranscriptSummarizer(Tool):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+        self.summarizer = pipeline("summarization", model="facebook/usin")
         self.api_url = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
         self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
 
@@ -27,12 +27,30 @@ class TranscriptSummarizer(Tool):
 
     def forward(self, transcript: str) -> str:
         try:
-            summary = self.summarizer(transcript, max_length=2000, min_length=750, do_sample=False)[0]['summary_text']
-            key_entities = summary.split()[:100]  # Extract the first 100 words
-            image_prompt = f"Generate an image related to: {' '.join(key_entities)}, professional style"
+            transcript_length = len(transcript)
+
+            def get_summary_lengths(length):
+                if length <= 1000:
+                    max_length = 300
+                    min_length = 100
+                elif length <= 3000:
+                    max_length = 750
+                    min_length = 250
+                else:
+                    max_length = 1500
+                    min_length = 500
+                return max_length, min_length
+
+            max_length, min_length = get_summary_lengths(transcript_length)
+            summary = self.summarizer(transcript, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
+            key_entities = summary.split()[:3]  # Extract first 3 words as key entities
+            image_prompt = f"Generate an image related to: {' '.join(key_entities)}, cartoon style"
             image_bytes = self.query({"inputs": image_prompt})
             image = Image.open(io.BytesIO(image_bytes))
-            image_url = "Images/image.jpg"  # Placeholder, as we can't directly pass PIL Image to Gradio
+            image_folder = "Image"
+            if not os.path.exists(image_folder):
+                os.makedirs(image_folder)
+            image_url = os.path.join(image_folder, "image.jpg")  # Specify the folder path
             image.save(image_url) # Save the image to a file
             return f"{summary}\n\nImage URL: {image_url}" # Return the file path
         except Exception as e: