alpha version 1.1

app.py

@@ -27,7 +27,19 @@ class MarketUseCaseGen:
                 progress_bar.progress(100)
                 # Display the generated article
                 st.subheader("Generated Article:")
+                
         if st.session_state.article_text and st.session_state.article_text != "":
+
+            # Convert the article text to a string if it's not already
+            if isinstance(st.session_state.article_text, str):
+                article_text = st.session_state.article_text
+            
+            elif isinstance(st.session_state.article_text, dict) and 'article' in st.session_state.article_text:
+                article_text = st.session_state.article_text['article']
+            
+            else:
+                article_text = str(st.session_state.article_text) 
+            
             with st.container():
             # Display the article text
                 st.markdown(st.session_state.article_text)
@@ -38,7 +50,7 @@ class MarketUseCaseGen:
                     if st.session_state.article_text:
                         st.download_button(
                             label="Download Article",
-                            data=st.session_state.article_text,
+                            data=article_text,
                             file_name=f"{st.session_state.name.replace(' ', '_').lower()}_market_and_use_case_analysis.txt",
                             mime="text/plain"
                         )

src/research_agent/config/tasks.yaml

@@ -5,7 +5,7 @@ research_task:
     - Find out the {company} strategic focus areas where it outperforms other companys in the same Industry
     - For additional information about the company use the following link: {company_link}
     IMPORTANT INSTRUCTIONS ABOUT USING TOOLS: When using tools, DO NOT ESCAPE the underscore character "_", EVER. If you need to use a tool and pass in a parameter called 'search_query', you should write 'search_query', not 'search\_query'. THIS IS VERY IMPORTANT, else the tool will not work.
-
+    Search a particular website takes the argument as search_query: 'string', website: 'string' in order to use the tool properly
   expected_output: >
     - A list with the most relevant information about industry,segment,key offering and strategic focus of {company} in a markdown file. 
     - For each source or contextual detail referenced, please include any relevant links in a markdown link format (e.g., [source text](URL)).
@@ -23,10 +23,10 @@ design_task:
     For flowcharts use the mermaid js library. 
     Ensure that all relevant information is accessible within the file, and integrate each link in context as necessary for clarity (e.g., [source text](URL)).
     IMPORTANT INSTRUCTIONS ABOUT USING TOOLS: When using tools, DO NOT ESCAPE the underscore character "_", EVER. If you need to use a tool and pass in a parameter called 'search_query', you should write 'search_query', not 'search\_query'. THIS IS VERY IMPORTANT, else the tool will not work.
-
+    Search a particular website takes the argument as search_query: 'string', website: 'string' in order to use the tool properly
   expected_output: >
     A markdown document with all the ideas to that can be implemented.
-    
+
     Each one should contain:
     - Title:
     - Summary: 
@@ -36,28 +36,26 @@ design_task:
     - How the idea may impact or benefit the customers
     - Source: [source text](URL)
 
-    
   agent: design_thinker
 
 resource_collector:
-    description: >
-      Based on the ideas generated by the design thinker, gather relevant resources to support each idea. 
-        - The resources should include relevant datasets, research papers, similar or existing projects, and any related documentation that may aid in exploring or validating the proposed expansion ideas.
-        - Each resource should be annotated with a short description explaining its relevance.
-        - If applicable, propose GenAI solutions like document search, automated report generation, and AI-powered chat systems for internal or customer-facing purposes.
-        - Datasets and existing projects can be found on websites such as hugging_face, github, kaggle.
-      
-      IMPORTANT INSTRUCTIONS ABOUT USING TOOLS: When using tools, DO NOT ESCAPE the underscore character "_", EVER. If you need to use a tool and pass in a parameter called 'search_query', you should write 'search_query', not 'search\_query'. THIS IS VERY IMPORTANT, else the tool will not work.
-    
-    Context: ["analysing_task"]
-
-    expected_output: >
-      A structured list of resources for any two ideas, with the following details for each resource:
-        - Title and type of the resource (e.g., Dataset, Research Paper, Project)
-        - Direct link or reference to the resource in markdown link format (e.g., [Title](URL))
-        - A short summary of how each resource is relevant to the specific idea.
-        - Modification to suit {company}'s goals 
-
-      Ensure that this information is formatted as markdown, with each resource link integrated in context for readability.
-
-    agent: developer
+  description: >
+    Based on the ideas generated by the design thinker, gather relevant resources to support each idea. 
+      - The resources should include relevant datasets, research papers, similar or existing projects, and any related documentation that may aid in exploring or validating the proposed expansion ideas.
+      - Each resource should be annotated with a short description explaining its relevance.
+      - If applicable, propose GenAI solutions like document search, automated report generation, and AI-powered chat systems for internal or customer-facing purposes.
+      - Datasets and existing projects can be found on websites such as hugging_face, github, kaggle.
+
+    IMPORTANT INSTRUCTIONS ABOUT USING TOOLS: When using tools, DO NOT ESCAPE the underscore character "_", EVER. If you need to use a tool and pass in a parameter called 'search_query', you should write 'search_query', not 'search\_query'. THIS IS VERY IMPORTANT, else the tool will not work.
+    Search a particular website takes the argument as search_query: 'string', website: 'string' in order to use the tool properly
+  Context: ["analysing_task"]
+  expected_output: >
+    A structured list of resources for any two ideas, with the following details for each resource:
+      - Title and type of the resource (e.g., Dataset, Research Paper, Project)
+      - Direct link or reference to the resource in markdown link format (e.g., [Title](URL))
+      - A short summary of how each resource is relevant to the specific idea.
+      - Modification to suit {company}'s goals 
+
+    Ensure that this information is formatted as markdown, with each resource link integrated in context for readability.
+
+  agent: developer