Update main.py

main.py

@@ -1,7 +1,7 @@
+import asyncio
 from crewai import Crew
 from textwrap import dedent
 import json
-
 from crypto_analysis_agents import CryptoAnalysisAgents
 from crypto__analysis_tasks import CryptoAnalysisTasks
 
@@ -9,7 +9,7 @@ class CryptoCrew:
     def __init__(self, crypto):
         self.crypto = crypto
 
-    def run(self):
+    async def run(self):
         agents = CryptoAnalysisAgents()
         tasks = CryptoAnalysisTasks()
 
@@ -35,63 +35,30 @@ class CryptoCrew:
                 recommend_task
             ],
             verbose=True,
-            iteration_limit=200,
-            time_limit=400
+            iteration_limit=300,
+            time_limit=600
         )
 
-        result = crew.kickoff()
+        try:
+            result = await asyncio.wait_for(crew.kickoff(), timeout=600)
+        except asyncio.TimeoutError:
+            return {"summary": "Analysis timed out. Please try again with a simpler query or increased time limit."}
 
-        # Attempt to parse the result based on its actual structure
         parsed_result = self.parse_result(result)
-
         return parsed_result
 
     def parse_result(self, result):
-        # Attempt to extract a text summary from the result
-        # Here, assume result has a method or attribute to get text (like result.get_text(), result.summary, etc.)
-        # Replace 'result.get_summary()' with the correct method/attribute after inspecting the result structure
-        try:
-            result_text = result.get_summary()  # Hypothetical method; replace with actual method/attribute
-        except AttributeError:
-            # If the method or attribute doesn't exist, fall back to another way
-            result_text = str(result)  # Convert to string if necessary
-
-        # Initialize the parsed result structure
-        parsed = {
-            "summary": result_text,
-            "sentiment": {
-                "overall": "Neutral",
-                "social_media": "Neutral",
-                "news": "Neutral",
-                "community": "Neutral"
-            }
-        }
-
-        # Perform sentiment extraction on the result_text
-        if "sentiment score" in result_text.lower():
-            sentiment_section = result_text.split("sentiment score")[1].split("\n")[0]
-            try:
-                sentiment_score = float(sentiment_section.strip())
-                if sentiment_score > 0.5:
-                    parsed["sentiment"]["overall"] = "Positive"
-                elif sentiment_score < -0.5:
-                    parsed["sentiment"]["overall"] = "Negative"
-            except ValueError:
-                pass  # Handle the case where conversion to float fails
-
-        return parsed
+        # ... (rest of the parse_result method remains the same)
 
 if __name__ == "__main__":
     print("## Welcome to Crypto Analysis Crew")
     print('-------------------------------')
-    crypto = input(
-        dedent("""
-        What is the cryptocurrency you want to analyze?
-        """))
-
+    crypto = input(dedent("""
+    What is the cryptocurrency you want to analyze?
+    """))
     crypto_crew = CryptoCrew(crypto)
-    result = crypto_crew.run()
+    result = asyncio.run(crypto_crew.run())
     print("\n\n########################")
     print("## Here is the Report")
     print("########################\n")
-    print(json.dumps(result, indent=2))
+    print(json.dumps(result, indent=2))