Update app.py

app.py

@@ -26,12 +26,9 @@ def process_event(event):
         base_url="https://api.deepinfra.com/v1/openai",
     )
 
-    
     llm_prompt = f""" 
-    
-    You are a digital marketing campaign analyst designed to analyze and report digital marketing campaign data for Rod Wave concerts, Your job is to convert the given text into JSON.
-    Dont make Any Assumptions,if value doesnt exist,Consider it as Zero.
-
+    You are a digital marketing campaign analyst designed to analyze and report digital marketing campaign data for Rod Wave concerts. Your job is to convert the given text into JSON.
+    Don't make any assumptions; if a value doesn't exist, consider it as zero.
     {{
     "market": "str",
     "total_spend": "float",
@@ -52,20 +49,27 @@ def process_event(event):
     "revenue_total_revenue": "float",
     "revenue_roi": "float"
     }}
-
-    Here is Text for it:
-
+    Here is the text for it:
     {event}
-
-
-    Return in only JSON adhering to Above Schema
+    Return in only JSON adhering to the above schema.
     """
     
-    chat_completion = openai.chat.completions.create(
-        model="Qwen/Qwen2.5-Coder-32B-Instruct",
-        messages=[{"role": "user", "content": llm_prompt}],
-    )
-    return chat_completion.choices[0].message.content
+    # Attempt to process the event and validate JSON
+    for attempt in range(2):  # Try twice
+        chat_completion = openai.chat.completions.create(
+            model="Qwen/Qwen2.5-Coder-32B-Instruct",
+            messages=[{"role": "user", "content": llm_prompt}],
+        )
+        json_output = chat_completion.choices[0].message.content
+        
+        try:
+            return extract_and_parse_json_from_markdown(json_output)
+        except ValueError:
+            if attempt == 0:
+                st.warning("JSON validation failed, retrying...")
+            else:
+                st.error("Failed to validate JSON after retrying.")
+                return None  # Return None if it fails after retrying
 
 def process_all_events(events):
     json_all = []
@@ -77,12 +81,10 @@ def process_all_events(events):
         for i, future in enumerate(concurrent.futures.as_completed(futures)):
             progress = (i + 1) / len(events)
             progress_bar.progress(progress)
-            # st.write(f"Processed event {i + 1}/{len(events)}")
             json_all.append(future.result())
     
     return json_all
 
-
 def main():
     st.title("Rod Wave Concert Marketing Data Processor")
     
@@ -100,7 +102,7 @@ def main():
         events = re.split(r'\n(?=Rod Wave Concert)', text)
         events = [event for event in events if event.strip()]
         
-        st.write(f"Found {len(events)} events to process")
+        st.write(f"Found **{len(events)}** events to process")
         
         if st.button("Process Data"):
             with st.spinner("Processing events..."):
@@ -108,7 +110,10 @@ def main():
                 
                 json_sanity = []
                 for ele in json_all:
-                    json_sanity.append(extract_and_parse_json_from_markdown(ele))
+                    if ele is not None:  # Only process valid JSON
+                        json_sanity.append(ele)
+                        st.subheader("Processed JSON for Event")
+                        st.json(ele)
                 
                 df = pd.DataFrame(json_sanity)