Spaces:
Runtime error
Runtime error
update app.py
Browse files
app.py
CHANGED
@@ -54,9 +54,18 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
|
|
54 |
with open("prompts.yaml", 'r') as stream:
|
55 |
prompt_templates = yaml.safe_load(stream)
|
56 |
|
|
|
57 |
@tool
|
58 |
def summarize_text(content: str) -> str:
|
59 |
-
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
try:
|
61 |
from transformers import pipeline
|
62 |
summarizer = pipeline("summarization")
|
@@ -65,9 +74,18 @@ def summarize_text(content: str) -> str:
|
|
65 |
except Exception as e:
|
66 |
return f"Error processing summary: {str(e)}"
|
67 |
|
|
|
68 |
@tool
|
69 |
def analyze_sentiment(text: str) -> str:
|
70 |
-
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
71 |
from textblob import TextBlob
|
72 |
sentiment_score = TextBlob(text).sentiment.polarity
|
73 |
if sentiment_score > 0:
|
@@ -77,6 +95,7 @@ def analyze_sentiment(text: str) -> str:
|
|
77 |
else:
|
78 |
return "Neutral sentiment detected."
|
79 |
|
|
|
80 |
# Ensure image generation is included
|
81 |
tools_list = [final_answer, image_generation_tool, summarize_text, analyze_sentiment, get_current_time_in_timezone]
|
82 |
|
|
|
54 |
with open("prompts.yaml", 'r') as stream:
|
55 |
prompt_templates = yaml.safe_load(stream)
|
56 |
|
57 |
+
|
58 |
@tool
|
59 |
def summarize_text(content: str) -> str:
|
60 |
+
"""
|
61 |
+
Summarizes a given text.
|
62 |
+
|
63 |
+
Args:
|
64 |
+
content (str): The text content that needs summarization.
|
65 |
+
|
66 |
+
Returns:
|
67 |
+
str: A summarized version of the input text.
|
68 |
+
"""
|
69 |
try:
|
70 |
from transformers import pipeline
|
71 |
summarizer = pipeline("summarization")
|
|
|
74 |
except Exception as e:
|
75 |
return f"Error processing summary: {str(e)}"
|
76 |
|
77 |
+
|
78 |
@tool
|
79 |
def analyze_sentiment(text: str) -> str:
|
80 |
+
"""
|
81 |
+
Analyzes the sentiment of a given text.
|
82 |
+
|
83 |
+
Args:
|
84 |
+
text (str): The input text for sentiment analysis.
|
85 |
+
|
86 |
+
Returns:
|
87 |
+
str: A sentiment classification ('Positive', 'Negative', or 'Neutral').
|
88 |
+
"""
|
89 |
from textblob import TextBlob
|
90 |
sentiment_score = TextBlob(text).sentiment.polarity
|
91 |
if sentiment_score > 0:
|
|
|
95 |
else:
|
96 |
return "Neutral sentiment detected."
|
97 |
|
98 |
+
|
99 |
# Ensure image generation is included
|
100 |
tools_list = [final_answer, image_generation_tool, summarize_text, analyze_sentiment, get_current_time_in_timezone]
|
101 |
|