deploy at 2024-08-21 11:25:05.598125

main.py

@@ -38,7 +38,7 @@ class Event(BaseModel):
     sentiment: Literal["Positive", "Negative"] = Field(..., description="Categorization of the event sentiment")
 
 class EventResponse(BaseModel):
-    events: List[Event] = Field(min_length=10, max_length=20, description="List of events extracted from the context")
+    events: List[Event] = Field(min_length=5, max_length=30, description="List of events extracted from the context")
 
 # Set up the Pydantic output parser
 parser = PydanticOutputParser(pydantic_object=EventResponse)
@@ -71,7 +71,7 @@ event_prompt = PromptTemplate(
 
 # Function to get the appropriate language model based on user selection
 def getModel(model, key):
-    if(model == 'OpenAI Gpt-o'):
+    if(model == 'OpenAI Gpt-4o'):
         os.environ['OPENAI_API_KEY'] = key
         return ChatOpenAI(temperature=0,  # Set to 0 for deterministic output
                     model="gpt-4o-2024-08-06",  # Using the GPT-4 Turbo model
@@ -169,7 +169,7 @@ def generate_timeline(topic, numevents, llm):
 #    wiki_content = page.content
 
     
-    wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=5000))
+    wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper(top_k_results=3, doc_content_chars_max=5000))
     wiki_content = wikipedia.run(topic)
 
     print(f"wiki_content: {wiki_content}")
@@ -193,7 +193,7 @@ def generate_timeline(topic, numevents, llm):
         df = get_timeline_df(result)
                 
         # Optionally, save the DataFrame to a CSV file
-        df.to_csv(f"timeline.csv", index=True)
+        df.to_csv(f"{topic.replace(' ', '_')}_timeline.csv", index=True)
         print("Results saved to 'results.csv'")    
     except Exception as e:
         print(f"Error parsing LLM output: {str(e)}")

ms_dhoni_timeline.csv

@@ -0,0 +1,21 @@
+,index,Time,Event,Sentiment,TimeStr
+0,0,1981-07-07,Mahendra Singh Dhoni is born,Positive,07/07/1981
+1,1,1999-01-01,Dhoni makes his first-class debut for Bihar,Positive,01/01/1999
+2,2,2004-12-23,Dhoni debuts for Indian cricket team in ODI against Bangladesh,Positive,23/12/2004
+3,3,2005-12-01,Dhoni plays his first Test match against Sri Lanka,Positive,01/12/2005
+4,4,2007-01-01,Dhoni becomes captain of India's ODI team,Positive,01/01/2007
+5,5,2007-09-24,Dhoni leads India to victory in ICC World Twenty20,Positive,24/09/2007
+6,6,2008-01-01,Dhoni takes over captaincy in all cricket formats for India,Positive,01/01/2008
+7,7,2008-01-01,Dhoni awarded Major Dhyan Chand Khel Ratna Award,Positive,01/01/2008
+8,8,2009-01-01,"Dhoni receives Padma Shri, fourth highest civilian award",Positive,01/01/2009
+9,10,2010-01-01,Dhoni leads Chennai Super Kings to first IPL victory,Positive,01/01/2010
+10,9,2010-01-01,Dhoni leads India to victory in Asia Cup,Positive,01/01/2010
+11,12,2011-01-01,Dhoni awarded honorary rank of Lieutenant Colonel in Indian Territorial Army,Positive,01/01/2011
+12,11,2011-04-02,Dhoni leads India to Cricket World Cup victory,Positive,02/04/2011
+13,13,2013-06-23,Dhoni leads India to ICC Champions Trophy victory,Positive,23/06/2013
+14,14,2014-12-30,Dhoni retires from Test cricket,Negative,30/12/2014
+15,15,2016-09-30,Biopic 'M.S. Dhoni: The Untold Story' released,Positive,30/09/2016
+16,16,2017-01-01,Dhoni steps down as captain in limited-overs formats,Negative,01/01/2017
+17,17,2018-01-01,"Dhoni receives Padma Bhushan, third highest civilian award",Positive,01/01/2018
+18,18,2019-01-01,Dhoni retires from limited-overs international cricket,Negative,01/01/2019
+19,19,2023-01-01,Dhoni leads Chennai Super Kings to fifth IPL victory,Positive,01/01/2023

space_exploration_timeline.csv

@@ -0,0 +1,12 @@
+,index,Time,Event,Sentiment,TimeStr
+0,0,1957-10-04,"The Soviet Union launched Sputnik 1, the first human-made object to orbit Earth.",Positive,04/10/1957
+1,1,1961-04-12,Yuri Gagarin became the first human in space aboard Vostok 1.,Positive,12/04/1961
+2,3,1965-03-18,Alexei Leonov conducted the first spacewalk.,Positive,18/03/1965
+3,4,1966-01-01,The first automatic landing on another celestial body was achieved.,Positive,01/01/1966
+4,2,1969-07-20,American Apollo 11 mission achieved the first Moon landing.,Positive,20/07/1969
+5,5,1971-01-01,"The first space station, Salyut 1, was launched.",Positive,01/01/1971
+6,6,2011-03-01,The International Space Station (ISS) was substantially completed.,Positive,01/03/2011
+7,7,2020-01-01,SpaceX's Starlink became the largest-ever satellite constellation.,Positive,01/01/2020
+8,8,2022-01-01,The first mission of the Artemis Program took place.,Positive,01/01/2022
+9,9,2023-04-01,"SpaceX's Starship, the largest and most powerful rocket, had its first flight.",Positive,01/04/2023
+10,10,2024-06-01,"SpaceX's Starship completed its fourth test flight, reaching space and returning both stages.",Positive,01/06/2024

timeline.csv

@@ -1,13 +1,13 @@
 ,index,Time,Event,Sentiment,TimeStr
-0,1,1957-01-01,Soviet space program puts first living being in orbit.,Positive,01/01/1957
-1,0,1957-10-04,"Soviet Union launches Sputnik 1, first human-made object to orbit Earth.",Positive,04/10/1957
-2,2,1961-01-01,Yuri Gagarin achieves first human spaceflight aboard Vostok 1.,Positive,01/01/1961
-3,3,1965-03-18,Alexei Leonov conducts first spacewalk in history.,Positive,18/03/1965
-4,4,1966-01-01,First automatic landing on another celestial body achieved.,Positive,01/01/1966
-5,5,1969-07-20,American Apollo 11 mission lands on the Moon.,Positive,20/07/1969
-6,6,1971-01-01,"Launch of first space station, Salyut 1.",Positive,01/01/1971
-7,10,2000-01-01,China initiates successful crewed spaceflight program.,Positive,01/01/2000
-8,11,2000-01-01,India launches Chandrayaan programme for space exploration.,Positive,01/01/2000
-9,8,2009-01-01,Expert panel judges Constellation program unrealistic.,Negative,01/01/2009
-10,7,2011-03-01,Substantial completion of International Space Station following STS-133.,Positive,01/03/2011
-11,9,2022-01-01,First Artemis Program mission occurs.,Positive,01/01/2022
+0,1,1957-01-01 00:00:00+00:00,Soviet Union sends first living being into orbit.,Positive,01/01/1957
+1,0,1957-10-04 00:00:00+00:00,"Soviet Union launches Sputnik 1, the first human-made object to orbit Earth.",Positive,04/10/1957
+2,2,1961-01-01 00:00:00+00:00,Yuri Gagarin aboard Vostok 1 becomes the first human in space.,Positive,01/01/1961
+3,3,1965-03-18 00:00:00+00:00,Alexei Leonov performs the first spacewalk.,Positive,18/03/1965
+4,4,1966-01-01 00:00:00+00:00,First automatic landing on another celestial body achieved.,Positive,01/01/1966
+5,5,1969-07-20 00:00:00+00:00,American Apollo 11 mission lands on the Moon.,Positive,20/07/1969
+6,6,1971-01-01 00:00:00+00:00,"Launch of Salyut 1, the first space station.",Positive,01/01/1971
+7,10,2000-01-01 00:00:00+00:00,China initiates successful crewed spaceflight program.,Positive,01/01/2000
+8,11,2000-01-01 00:00:00+00:00,India launches Chandrayaan programme for lunar exploration.,Positive,01/01/2000
+9,8,2009-01-01 00:00:00+00:00,Expert review panel judges Constellation program unrealistic.,Negative,01/01/2009
+10,7,2011-03-01 00:00:00+00:00,Substantial completion of the International Space Station following STS-133.,Positive,01/03/2011
+11,9,2022-01-01 00:00:00+00:00,First Artemis Program mission occurs.,Positive,01/01/2022