Update app.py

app.py

@@ -26,11 +26,52 @@ with gr.Blocks(theme=gr.themes.Monochrome(),css="footer{display:none !important}
     
     Chatbot = gr.Chatbot(label="Anonymous User")
     
+    g = Github(token)
+
+    def predict_prompt(input):
+        message_history.append({"role": "user", "content": input})
+        create_prompt = openai.ChatCompletion.create(
+            model = "gpt-4o",
+            messages = message_history
+        )
+
+        # Reply content for the chat bot
+        reply_prompt = create_prompt.choices[0].message.content
+
+        # Append answer as assistant reply to keep history of prompts
+        message_history.append({"role": "assistant", "content": reply_prompt}) 
+        response = [(message_history[i]["content"], message_history[i+1]["content"]) for i in range(1, len(message_history) - 1, 2)]
+
+        # Add token and connect to repository (again)
+        g = Github(token)
+
+        # Repository details
+        repo = g.get_user().get_repo("CATData")
+        file_path = filename
+        commit_message = "Updating file content"
+
+        # variable content store each message found in the message history and states the respective user. This is for the content to add in the file
+        content = ''
+        for i in range(1,len(message_history), 1):
+            if (i % 2 == 1) and (message_history[i]["role"] == "user"):
+                content += ('User: ' + message_history[i]["content"] + '\n')
+            if (i % 2 == 0)  and (message_history[i]["role"] == "assistant"):
+                content += ('Bot: ' + message_history[i]["content"] + '\n')
+            else:
+                content += ''
+
+        #updating the repository with the entire message history
+        file = repo.get_contents(file_path)
+        repo.update_file(file.path, commit_message, content, file.sha)
+
+        #return response
+        return response
+    
     with gr.Row():
         txt = gr.Textbox(
             show_label=False,
             placeholder = initial_message)
-        txt.submit(None, txt, Chatbot)    
+        txt.submit(predict_prompt, txt, Chatbot)    
         txt.submit(None, None, txt, js="() => {''}")
             
 chatblock.launch()