Update main.py

main.py

@@ -1,88 +1,88 @@
-from crewai import Crew
-from textwrap import dedent
-import json
-
-from crypto_analysis_agents import CryptoAnalysisAgents
-from crypto_analysis_tasks import CryptoAnalysisTasks
-
-class CryptoCrew:
-    def __init__(self, crypto):
-        self.crypto = crypto
-
-    def run(self):
-        agents = CryptoAnalysisAgents()
-        tasks = CryptoAnalysisTasks()
-
-        market_analyst_agent = agents.market_analyst()
-        technical_analyst_agent = agents.technical_analyst()
-        crypto_advisor_agent = agents.crypto_advisor()
-
-        market_research_task = tasks.market_research(market_analyst_agent, self.crypto)
-        technical_analysis_task = tasks.technical_analysis(technical_analyst_agent)
-        sentiment_analysis_task = tasks.sentiment_analysis(market_analyst_agent)
-        recommend_task = tasks.recommend(crypto_advisor_agent)
-
-        crew = Crew(
-            agents=[
-                market_analyst_agent,
-                technical_analyst_agent,
-                crypto_advisor_agent
-            ],
-            tasks=[
-                market_research_task,
-                technical_analysis_task,
-                sentiment_analysis_task,
-                recommend_task
-            ],
-            verbose=True
-        )
-
-        result = crew.kickoff()
-        
-        # Parse the result to extract structured data
-        parsed_result = self.parse_result(result)
-        
-        return parsed_result
-
-    def parse_result(self, result):
-        # This is a simple parser. You might need to adjust it based on the actual structure of your result
-        parsed = {
-            "summary": result,
-            "sentiment": {
-                "overall": "Neutral",
-                "social_media": "Neutral",
-                "news": "Neutral",
-                "community": "Neutral"
-            }
-        }
-        
-        # Extract sentiment information
-        if "sentiment score" in result.lower():
-            sentiment_section = result.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:
-                pass
-
-        # You can add more parsing logic here to extract other structured data
-
-        return parsed
-
-if __name__ == "__main__":
-    print("## Welcome to Crypto Analysis Crew")
-    print('-------------------------------')
-    crypto = input(
-        dedent("""
-        What is the cryptocurrency you want to analyze?
-        """))
-    
-    crypto_crew = CryptoCrew(crypto)
-    result = crypto_crew.run()
-    print("\n\n########################")
-    print("## Here is the Report")
-    print("########################\n")
+from crewai import Crew
+from textwrap import dedent
+import json
+
+from crypto_analysis_agents import CryptoAnalysisAgent
+from crypto_analysis_tasks import CryptoAnalysisTasks
+
+class CryptoCrew:
+    def __init__(self, crypto):
+        self.crypto = crypto
+
+    def run(self):
+        agents = CryptoAnalysisAgent()
+        tasks = CryptoAnalysisTasks()
+
+        market_analyst_agent = agents.market_analyst()
+        technical_analyst_agent = agents.technical_analyst()
+        crypto_advisor_agent = agents.crypto_advisor()
+
+        market_research_task = tasks.market_research(market_analyst_agent, self.crypto)
+        technical_analysis_task = tasks.technical_analysis(technical_analyst_agent)
+        sentiment_analysis_task = tasks.sentiment_analysis(market_analyst_agent)
+        recommend_task = tasks.recommend(crypto_advisor_agent)
+
+        crew = Crew(
+            agents=[
+                market_analyst_agent,
+                technical_analyst_agent,
+                crypto_advisor_agent
+            ],
+            tasks=[
+                market_research_task,
+                technical_analysis_task,
+                sentiment_analysis_task,
+                recommend_task
+            ],
+            verbose=True
+        )
+
+        result = crew.kickoff()
+        
+        # Parse the result to extract structured data
+        parsed_result = self.parse_result(result)
+        
+        return parsed_result
+
+    def parse_result(self, result):
+        # This is a simple parser. You might need to adjust it based on the actual structure of your result
+        parsed = {
+            "summary": result,
+            "sentiment": {
+                "overall": "Neutral",
+                "social_media": "Neutral",
+                "news": "Neutral",
+                "community": "Neutral"
+            }
+        }
+        
+        # Extract sentiment information
+        if "sentiment score" in result.lower():
+            sentiment_section = result.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:
+                pass
+
+        # You can add more parsing logic here to extract other structured data
+
+        return parsed
+
+if __name__ == "__main__":
+    print("## Welcome to Crypto Analysis Crew")
+    print('-------------------------------')
+    crypto = input(
+        dedent("""
+        What is the cryptocurrency you want to analyze?
+        """))
+    
+    crypto_crew = CryptoCrew(crypto)
+    result = crypto_crew.run()
+    print("\n\n########################")
+    print("## Here is the Report")
+    print("########################\n")
     print(json.dumps(result, indent=2))