Update app.py

app.py

@@ -56,20 +56,34 @@ workflow.add_edge("researcher", "chart_generator")
 workflow.add_edge("chart_generator", END)
 graph = workflow.compile()
 
+import re
+
 def extract_chart_data(text):
     """
-    Try to extract something like:
-    2018: 20
-    2019: 21.5
-    2020: 18
+    Extracts year-value pairs from text, supporting formats like:
+    - 2019: 21.5
+    - 2020 - $18.2
+    - GDP in 2021 was 20.5 trillion USD
     """
-    import re
-    matches = re.findall(r'(\d{4})\s*[:\-]?\s*\$?([\d\.]+)', text)
+    matches = re.findall(r'(\b19\d{2}|\b20\d{2})[^\d]{1,10}(\$?\d+(\.\d+)?)', text)
+    
     if not matches:
         return None, None
-    years = [m[0] for m in matches]
-    values = [float(m[1]) for m in matches]
-    return years, values
+
+    years = []
+    values = []
+    for match in matches:
+        year = match[0]
+        value_str = match[1].replace('$', '')
+        try:
+            value = float(value_str)
+            years.append(year)
+            values.append(value)
+        except ValueError:
+            continue
+
+    return years, values if years and values else (None, None)
+
 
 def generate_plot(years, values):
     fig, ax = plt.subplots()
@@ -97,7 +111,9 @@ def run_langgraph(user_input):
 
 def process_input(user_input):
     result_text = run_langgraph(user_input)
+    print("LLM Output:", result_text)
     years, values = extract_chart_data(result_text)
+    print("Extracted Data:", years, values)
 
     if years and values:
         chart = generate_plot(years, values)
@@ -107,6 +123,7 @@ def process_input(user_input):
 
 
 
+
 interface = gr.Interface(
     fn=process_input,
     inputs="text",