first push

app.py

@@ -0,0 +1,157 @@
+import os
+# os.system('pip install requests')
+import requests
+
+def gpt3_question(prompt):
+    api_endpoint = "https://api.openai.com/v1/engines/text-davinci-003/completions"
+    api_key = "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"
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+    data = {
+        "prompt": prompt,
+        "max_tokens": 400,
+        "temperature": 0.5
+    }
+    print('sending request')
+    response = requests.post(api_endpoint, headers=headers, json=data)
+    print(response)
+    generated_text = response.json()["choices"][0]["text"]
+
+    return generated_text
+
+
+def chatgpt3_question(prompt):
+
+    url = "https://api.openai.com/v1/chat/completions"
+    api_key = "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"
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    data = {
+        "model": "gpt-3.5-turbo",
+        "messages": [{"role": "user", "content": prompt}]
+    }
+
+    response = requests.post(url, headers=headers, json=data)
+    generated_text = response.json()['choices'][0]['message']['content']
+
+    return generated_text
+
+
+def history2prompt(history, extra):
+    # history = [('The other day it was raining, and while I was driving a hit a stranger with my car.', 'Did you stop and render aid to the victim after the accident?'), ('True', 'Did you kill the guy?'), ('False', 'Was he part of the Mafia?')]
+    history_ = [item for tup in history for item in tup]
+    history_.append(extra)
+    print(history_)
+
+    if len(history_) > 1:
+        combinations = []
+        for i in range(1, len(history_)):
+            if i % 2 == 1:
+                combinations.append([i, i+2])
+
+        history_full = list()
+        history_full.append(history_[0])
+        for range_ in combinations:
+            history_full.append(' - '.join(history_[range_[0]:range_[1]]))
+
+        return '\n'.join(history_full)
+    else:
+        return history_[0]
+
+# gpt3_keywords('The other day it was raining, and while I was driving a hit a stranger with my car.')
+
+import subprocess
+import random
+import gradio as gr
+import requests
+
+history = None
+history_prompt = None
+history_final = None
+block_predict = False
+block_advice = False
+
+def predict(input, history):
+    #WE CAN PLAY WITH user_input AND bot_answer, as well as history
+    user_input = input
+
+    # print('##', [x for x in history], input)
+    global history_prompt
+    global history_final
+    global block_predict
+
+    if block_predict == False:
+        print('@@@', history)
+        history_prompt = history2prompt(history, input)
+        print('###', history_prompt)
+
+        prompt = f"""
+        Imagine being a criminal lawyer being told the following story with the following circumstances: {history_prompt}
+        Output the first relevant legal question that can result in the highest incrimination for the client (if somebody is hurt, start from fatal injuries), and that can only be answered as Yes or No
+        """
+        bot_answer = gpt3_question(prompt)
+
+        response = list()
+        response = [(input, bot_answer)]
+        
+        history.append(response[0])
+        response = history
+        history_final = history
+
+        # print('#history', history)
+        # print('#response', response)
+
+        return response, history
+
+def chatbot_foo():
+    global history_prompt
+    global history_final
+    global block_predict
+    global block_advice
+
+    if block_advice == False:
+        
+        prompt = f"""
+        Imagine being an Ohio criminal lawyer being told the following story with the following circumstances: {history_prompt}
+        Tell the client how much does he risk in terms of criminal charges, prison, and cite sources from law books
+        """
+        bot_answer = gpt3_question(prompt)
+
+        history_final.append(('Consult me on the matter:', bot_answer))
+
+        block_predict = True
+        block_advice = True
+        return history_final, history_final
+
+demo = gr.Blocks()
+with demo:
+    gr.Markdown(
+        """
+    <center>    
+        Chat with Morty by typing in the input box below.
+    </center>
+    """
+    )
+    state = gr.Variable(value=[]) #beginning
+    chatbot = gr.Chatbot(color_map=("#00ff7f", "#00d5ff"))
+    text = gr.Textbox(
+        label="Talk to your lawyer (press enter to submit)",
+        value="The other day it was raining, and while I was driving a hit a stranger with my car.",
+        placeholder="What is your name?",
+        max_lines=1,
+    )
+    text.submit(predict, [text, state], [chatbot, state])
+    text.submit(lambda x: "", text, text)
+
+    btn = gr.Button(value="submit")
+    btn.click(chatbot_foo, None, [chatbot, state])
+    # true_false_radio = gr.Radio(choices=["True", "False"], label="Select True or False")
+    # iface = gr.Interface(fn=my_function, inputs=[text, true_false_radio], outputs=chatbot, live=True, capture_session=True)
+
+demo.launch(share=False)