push to production

app.py

@@ -11,6 +11,9 @@ from langchain_groq.chat_models import ChatGroq
 from langchain_google_genai import GoogleGenerativeAI
 from streamlit_feedback import streamlit_feedback
 from huggingface_hub import HfApi
+from datasets import load_dataset, get_dataset_config_info
+from PIL import Image
+
 st.set_page_config(layout="wide")
 
 # Load environment variables : Groq and Hugging Face API keys
@@ -271,6 +274,27 @@ import uuid
                     answer = f"!!!Faced an error while working on your query. Please try again!!!"
                 
                 response = {"role": "assistant", "content": answer, "gen_code": code, "ex_code": code, "last_prompt": prompt, "error": error}
+                
+                # Update to HuggingFace dataset
+                dataset_config = get_dataset_config_info("SustainabilityLabIITGN/VayuBuddy_logs", token=hf_token)
+                splits = dataset_config.splits
+                last_split = list(splits.keys())[-1]
+                last_split_size = splits[last_split].num_examples
+                
+                if last_split_size >= 100:
+                    current_split = str(int(last_split) + 1)
+                else:
+                    current_split = last_split
+                
+                try:
+                    img = Image.open(answer)
+                    answer = None
+                    image = answer
+                except:
+                    image = None
+                    
+                ds = ds.add_item({"prompt": prompt, "code": code, "answer": answer, "error": error, "model": model_name, "image": image})
+                ds.push_to_hub("SustainabilityLabIITGN/VayuBuddy_logs", split=current_split, token=hf_token)
 
                 # Get response from agent
                 # response = ask_question(model_name=model_name, question=prompt)
@@ -290,10 +314,10 @@ import uuid
 # contact details
 contact_details = """
 **Feel free to reach out to us:**
-- [Yash J Bachwana](mailto:[email protected])
-  (Lead Developer, IIT Gandhinagar)
 - [Zeel B Patel](https://patel-zeel.github.io/)
   (PhD Student, IIT Gandhinagar)
+- Yash J Bachwana
+  (Developer, IIT Gandhinagar)
 - [Nipun Batra](https://nipunbatra.github.io/)
   (Faculty, IIT Gandhinagar)
 """