Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from transformers import AutoTokenizer, TFBlenderbotForConditionalGeneration
 import tensorflow as tf
+import json
+
 print("Loading the model......")
 model_name = "WICKED4950/Irisonego5"
 strategy = tf.distribute.MirroredStrategy()
@@ -9,6 +11,16 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 with strategy.scope():
     model = TFBlenderbotForConditionalGeneration.from_pretrained(model_name)
 
+def save_question(path = "question_answer.json",question,answer):
+    try:
+        with open(path, "r") as file:
+            data = json.load(file)
+        data.append({"Question:":question,"Answer:":answer})
+        with open(path, "w") as file:
+            json.dump(data, file, indent=4)
+    except Exception as e:
+        print(f"Error with {e}")
+        
 print("Interface getting done....")
 # Define the chatbot function
 def predict(user_input):
@@ -27,6 +39,7 @@ def predict(user_input):
 
     # Decode the response
     response = tokenizer.decode(response_id[0], skip_special_tokens=True)
+    save_question(question = user_input,answer=response)
     return response
 
 # Gradio interface