Update app.py

app.py

@@ -1,15 +1,18 @@
 import gradio as gr
 import random
-import torch
-from transformers import AutoConfig, AutoTokenizer, AutoModelWithLMHead
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
-from itertools import chain
-
 import os 
 
+# THIS IS TABBED INTERFACE that calls other spaces
+## SEE https://huggingface.co/spaces/gorkemgoknar/movie_chat_gpt_yourtts_fileinput for file input
+## SEE https://huggingface.co/spaces/gorkemgoknar/movie_chat_gpt_yourtts for mic input
+## SEE https://huggingface.co/spaces/gorkemgoknar/moviechatbot for text input
+
+
+
 #emotion_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
 #emotion_model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
 
+# TODO voice based on emotion
 def get_emotion(text):
   input_ids = tokenizer.encode(text + '</s>', return_tensors='pt')
   output = model.generate(input_ids=input_ids,max_length=2)
@@ -17,61 +20,8 @@ def get_emotion(text):
   label = dec[0]
   return label.split()[1]
 
-    
-config = AutoConfig.from_pretrained('gorkemgoknar/gpt2chatbotenglish')
-model = GPT2LMHeadModel.from_pretrained('gorkemgoknar/gpt2chatbotenglish', config=config)
-
-tokenizer = GPT2Tokenizer.from_pretrained('gorkemgoknar/gpt2chatbotenglish')
-tokenizer.model_max_length = 1024
-
-#Dynamic Temperature 
-#See experiment https://www.linkedin.com/pulse/ai-goes-job-interview-g%25C3%25B6rkem-g%25C3%25B6knar
-
-base_temperature = 1.2
-dynamic_temperature_range = 0.15
-
-rand_range = random.uniform(-1 * dynamic_temperature_range , dynamic_temperature_range )
-temperature = base_temperature  + rand_range
-
-SPECIAL_TOKENS = ["<bos>", "<eos>", "<speaker1>", "<speaker2>", "<pad>"]
-
 #See document for experiment https://www.linkedin.com/pulse/ai-goes-job-interview-g%C3%B6rkem-g%C3%B6knar/
 
-def get_chat_response(name,history=[], input_txt = "Hello , what is your name?"):
-  
-  ai_history = history.copy()
-
-  #ai_history.append(input_txt)
-  ai_history_e = [tokenizer.encode(e) for e in ai_history]
-
-  personality = "My name is " + name
-
-  bos, eos, speaker1, speaker2 = tokenizer.convert_tokens_to_ids(SPECIAL_TOKENS[:-1])
-  
-  #persona first, history next, input text must be at the end
-  #[[bos, persona] , [history] , [input]]
-  sequence = [[bos]  + tokenizer.encode(personality)] + ai_history_e  + [tokenizer.encode(input_txt)]
-  ##[[bos, persona] , [speaker1 .., speakser2 .., speaker1 ... speaker2 ... , [input]]
-  sequence = [sequence[0]] + [[speaker2 if (len(sequence)-i) % 2 else speaker1] + s for i, s in enumerate(sequence[1:])]
-  
-  sequence = list(chain(*sequence))
-
-  #bot_input_ids = tokenizer.encode(personality + tokenizer.eos_token + input_txt + tokenizer.eos_token  , return_tensors='pt')
-  sequence_len = len(sequence)
-
-  #optimum response and speed
-  chat_history_ids = model.generate(
-          torch.tensor(sequence).unsqueeze(0), max_length=50,
-          pad_token_id=tokenizer.eos_token_id,  
-          no_repeat_ngram_size=3,       
-          do_sample=True, 
-          top_k=60, 
-          top_p=0.8,
-          temperature = 1.3
-      )
-  out_str = tokenizer.decode(chat_history_ids[0][sequence_len:],  skip_special_tokens=True)
-  #out_str = tokenizer.decode(chat_history_ids[:, sequence.shape[-1]:][0], skip_special_tokens=False)
-  return out_str
   
 ##you can use anyone from below
 '''
@@ -101,10 +51,7 @@ def get_chat_response(name,history=[], input_txt = "Hello , what is your name?")
 | Brother | Allnut | Rose | Qui-Gon | Jar Jar
 '''
 
-MODEL_NAME= "tts_models/multilingual/multi-dataset/your_tts"
-
 
-## SEE https://huggingface.co/spaces/gorkemgoknar/movie_chat_gpt_yourtts_fileinput for actual call
 
 
 
@@ -141,7 +88,7 @@ interface_file= gr.Interface.load("gorkemgoknar/movie_chat_gpt_yourtts_fileinput
                                 "text", 
                                 "state"], 
                         outputs=["html","state",gr.outputs.Audio(type="file")],      
-                        css=css, title=title, description=description,article=article )
+                        css=css, title="Chat with Uploaded file", description=description,article=article )
 
 
 interface_mic= gr.Interface.load("gorkemgoknar/movie_chat_gpt_yourtts",
@@ -151,7 +98,7 @@ interface_mic= gr.Interface.load("gorkemgoknar/movie_chat_gpt_yourtts",
                                 "text", 
                                 "state"], 
                         outputs=["html","state",gr.outputs.Audio(type="file")],      
-                        css=css, title=title, description=description,article=article )
+                        css=css, title="Chat with Your Voice", description=description,article=article )
 
 
 interface_text = gr.Interface.load("gorkemgoknar/moviechatbot",
@@ -162,6 +109,6 @@ interface_text = gr.Interface.load("gorkemgoknar/moviechatbot",
                         outputs=["html","state"],      
                         css=css, title="Chat Text Only", description=description,article=article)
             
-appinterface = gr.TabbedInterface([interface_mic, interface_text], ["Chat with Record",  "Chat Text only"])      
+appinterface = gr.TabbedInterface([interface_mic,interface_file, interface_text], ["Chat with Mic Record","Chat with Audio Upload" , "Chat Text only"])      
 if __name__ == "__main__":
     appinterface.launch()