Upload app.py

app.py

@@ -0,0 +1,54 @@
+import openai
+import numpy as np
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts import ChatPromptTemplate
+import gradio as gr
+import os
+def get_completion(relation, emotion, resolution, context):
+  openai_api_key = "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"
+  chat = ChatOpenAI(temperature=0.0, model='gpt-4', openai_api_key=openai_api_key)
+
+  template_string = '''
+Your task is to provide some recommendations to resolve a conflict between two {relation}. Below is more info:
+
+Emotional Impact: {emotion}
+Goal: {resolution}
+Context: {context}
+
+Please give some recommendations on how the two {relation} can reach their goal, given the emotional impact and context. Don't provide examples. Don't make general statements that apply in any situation. Please consider the typical environments where the {relation} interact in your answer and be specific on where/how exactly to go about resolving the conflict. Keep in mind the emotional impact - if the conflict is overwhelming then more serious measures may be needed. But if the emotional impact is low then your recommendations should be more casual - no need for a third party to get involved.
+
+Provide your response in a short paragraph, roughly 5-10 sentences.
+'''
+
+  prompt_template = ChatPromptTemplate.from_template(template_string)
+  message = prompt_template.format_messages(
+                      relation = relation,
+                      emotion = emotion,
+                      resolution = resolution,
+                      context = context
+                      )
+
+  response = chat(message)
+  return response.content
+
+relat = ['Co-workers', 'Family Members', 'Friends', 'Romantic partners']
+emot = ['Calm', 'Concerned', 'Annoyed', 'Upset', 'Overwhelmed']
+res = ['Compromise - Both parties make concessions to reach a mutual agreement. Useful when both sides have significant stakes and neither can fully satisfy their desires without the cooperation of the other.',
+       'Confrontation - Directly addressing the conflict head-on. This may be necessary when one side believes strongly in their stance and feels the need to stand firm.',
+       'Understanding - The goal is to better understand the perspective of the other party, without necessarily seeking to change or challenge it. This is particularly effective for clarifying miscommunications or misconceptions.',
+       'Avoidance - Sometimes it is better to postpone or sidestep a conflict if it is not critical, especially if the timing is not right or if more preparation is needed.',
+       'Accommodation - One party willingly yields to the desires of the other, perhaps because they deem the issue less significant for them or they prioritize the relationship over the specific disagreement.'
+      ]
+
+with gr.Blocks() as main:
+  mk = gr.Markdown(value='# AiOS Inter Accomplish')
+  relat_sel = gr.Dropdown(label='Relationship Category', choices=relat)
+  emot_sel = gr.Dropdown(label='Emotional Impact', choices=emot)
+  res_sel = gr.Dropdown(label='Resolution Type', choices=res)
+  context = gr.Textbox(value='', lines=5, label='Other Info')
+
+  sub = gr.Button(value='Submit')
+  ans = gr.TextArea(label='Response', value='')
+  out = sub.click(get_completion, inputs=[relat_sel, emot_sel, res_sel, context], outputs=ans, show_progress=True, scroll_to_output=True)
+
+main.launch(debug=True)