Update app.py

app.py

@@ -4,6 +4,7 @@ import openai
 
 from textblob import TextBlob
 from gtts import gTTS
+from pdfminer.high_level import extract_text
 
 openai.api_key = os.environ['OPENAI_API_KEY']
 
@@ -70,7 +71,27 @@ def siriGPT(audio):
     narrate_ans.save("narrate.wav") 
 
     return "narrate.wav"
+    
+
+def fileGPT(prompt, file_obj):
+    global messages 
 
+    file_text = extract_text(file_obj.name)
+    text = prompt.append(file_text)
+    
+    messages.append({"role": "user", "content": text})
+    
+    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
+
+    system_message = response["choices"][0]["message"]
+    messages.append(system_message)
+
+    chats = ""
+    for msg in messages:
+        if msg['role'] != 'system':
+            chats += msg['role'] + ": " + msg['content'] + "\n\n"
+
+    return chats
 
 
 
@@ -104,7 +125,8 @@ with gr.Blocks() as chatHistory:
 text = gr.Interface(fn=textGPT, inputs="text", outputs="text")
 audio = gr.Interface(fn=audioGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs="text")
 siri = gr.Interface(fn=siriGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs = "audio")
-demo = gr.TabbedInterface([text, audio, siri, chatHistory], [ "chatGPT", "audioGPT", "siriGPT", "ChatHistory"])
+file = gr.Interface(fn=fileGPT, inputs=["text", "file"], outputs="text", description = 'Enter prompt and your PDF, e.g. let's think step by step, summarize this following text:')
+demo = gr.TabbedInterface([text, audio, siri, file, chatHistory], [ "chatGPT", "audioGPT", "siriGPT", "fileGPT", "ChatHistory"])
 
 if __name__ == "__main__":
     demo.launch(enable_queue=False, auth=lambda u, p: user_db.get(u) == p,