Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -38,16 +38,25 @@ class CryptoCrew:
|
|
| 38 |
)
|
| 39 |
|
| 40 |
result = crew.kickoff()
|
| 41 |
-
|
|
|
|
| 42 |
parsed_result = self.parse_result(result)
|
| 43 |
-
|
| 44 |
return parsed_result
|
| 45 |
|
| 46 |
def parse_result(self, result):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
|
| 48 |
-
|
| 49 |
parsed = {
|
| 50 |
-
"summary":
|
| 51 |
"sentiment": {
|
| 52 |
"overall": "Neutral",
|
| 53 |
"social_media": "Neutral",
|
|
@@ -55,17 +64,18 @@ class CryptoCrew:
|
|
| 55 |
"community": "Neutral"
|
| 56 |
}
|
| 57 |
}
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
|
|
|
| 61 |
try:
|
| 62 |
sentiment_score = float(sentiment_section.strip())
|
| 63 |
if sentiment_score > 0.5:
|
| 64 |
parsed["sentiment"]["overall"] = "Positive"
|
| 65 |
elif sentiment_score < -0.5:
|
| 66 |
parsed["sentiment"]["overall"] = "Negative"
|
| 67 |
-
except:
|
| 68 |
-
pass
|
| 69 |
|
| 70 |
return parsed
|
| 71 |
|
|
@@ -76,10 +86,10 @@ if __name__ == "__main__":
|
|
| 76 |
dedent("""
|
| 77 |
What is the cryptocurrency you want to analyze?
|
| 78 |
"""))
|
| 79 |
-
|
| 80 |
crypto_crew = CryptoCrew(crypto)
|
| 81 |
result = crypto_crew.run()
|
| 82 |
print("\n\n########################")
|
| 83 |
print("## Here is the Report")
|
| 84 |
print("########################\n")
|
| 85 |
-
print(json.dumps(result, indent=2))
|
|
|
|
| 38 |
)
|
| 39 |
|
| 40 |
result = crew.kickoff()
|
| 41 |
+
|
| 42 |
+
# Attempt to parse the result based on its actual structure
|
| 43 |
parsed_result = self.parse_result(result)
|
| 44 |
+
|
| 45 |
return parsed_result
|
| 46 |
|
| 47 |
def parse_result(self, result):
|
| 48 |
+
# Attempt to extract a text summary from the result
|
| 49 |
+
# Here, assume result has a method or attribute to get text (like result.get_text(), result.summary, etc.)
|
| 50 |
+
# Replace 'result.get_summary()' with the correct method/attribute after inspecting the result structure
|
| 51 |
+
try:
|
| 52 |
+
result_text = result.get_summary() # Hypothetical method; replace with actual method/attribute
|
| 53 |
+
except AttributeError:
|
| 54 |
+
# If the method or attribute doesn't exist, fall back to another way
|
| 55 |
+
result_text = str(result) # Convert to string if necessary
|
| 56 |
|
| 57 |
+
# Initialize the parsed result structure
|
| 58 |
parsed = {
|
| 59 |
+
"summary": result_text,
|
| 60 |
"sentiment": {
|
| 61 |
"overall": "Neutral",
|
| 62 |
"social_media": "Neutral",
|
|
|
|
| 64 |
"community": "Neutral"
|
| 65 |
}
|
| 66 |
}
|
| 67 |
+
|
| 68 |
+
# Perform sentiment extraction on the result_text
|
| 69 |
+
if "sentiment score" in result_text.lower():
|
| 70 |
+
sentiment_section = result_text.split("sentiment score")[1].split("\n")[0]
|
| 71 |
try:
|
| 72 |
sentiment_score = float(sentiment_section.strip())
|
| 73 |
if sentiment_score > 0.5:
|
| 74 |
parsed["sentiment"]["overall"] = "Positive"
|
| 75 |
elif sentiment_score < -0.5:
|
| 76 |
parsed["sentiment"]["overall"] = "Negative"
|
| 77 |
+
except ValueError:
|
| 78 |
+
pass # Handle the case where conversion to float fails
|
| 79 |
|
| 80 |
return parsed
|
| 81 |
|
|
|
|
| 86 |
dedent("""
|
| 87 |
What is the cryptocurrency you want to analyze?
|
| 88 |
"""))
|
| 89 |
+
|
| 90 |
crypto_crew = CryptoCrew(crypto)
|
| 91 |
result = crypto_crew.run()
|
| 92 |
print("\n\n########################")
|
| 93 |
print("## Here is the Report")
|
| 94 |
print("########################\n")
|
| 95 |
+
print(json.dumps(result, indent=2))
|