Update chat.py

chat.py

@@ -249,7 +249,8 @@ class ChatHandler:
                     
                 else:
                     print("No table data found in the response.")
-                    visual_response = None
+                    global_df = unstructured_text_to_df(response)
+                    visual_response = create_visualization_csv(visual_query)
             else:
                 visual_response = None
             return response, visual_response
@@ -298,4 +299,58 @@ def convert_column_types(df):
             except ValueError:
                 # Leave as string if neither integer nor date
                 pass
-    return df    
+    return df    
+
+def unstructured_text_to_df(text):
+    import openai
+    import pandas as pd
+    import os
+    import json
+    
+    # Your OpenAI API key
+    openai.api_key = os.getenv("OPENAI_API_KEY", "")
+    
+    # Unstructured text
+    text = """The top 3 predicted materials in 2025 are:
+    1. 9762_NUT_CHNNL with a total quantity of 753
+    2. 8268_KIT_TOOL with a total quantity of 738
+    3. 5960_CABLE with a total quantity of 729"""
+    
+    # OpenAI prompt to structure the data
+    prompt = f"""
+    Extract the materials and their quantities from the following text and format them as a structured JSON:
+    {text}
+    """
+    
+    # Call OpenAI API
+    response = openai.chat.completions.create(
+    model="gpt-4o-mini",
+    messages=[
+        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "user", "content": prompt}
+    ],
+    temperature=0
+    )
+    
+    # Extract the response content
+    response_content = response.choices[0].message.content.strip()
+    
+    # Debugging: Print raw response to check its format
+    print("Raw Response:", response_content)
+    
+    # Try to parse the response as JSON
+    try:
+        structured_data = json.loads(response_content)  # Parse the JSON content
+        print("Parsed JSON:", structured_data)
+    except json.JSONDecodeError:
+        print("Error: Response content is not valid JSON.")
+    
+    
+    # Convert the structured data into a DataFrame
+    df = pd.DataFrame(structured_data["materials"])
+    
+    # Rename columns to desired format
+    df.columns = ["material_name", "material_quantity"]
+    
+    # Print the DataFrame
+    print(df)