Update app.py

app.py

@@ -36,9 +36,12 @@ h1 {
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
 
+
+
 def extract_text(path):
     return open(path, 'r').read()
 
+
 def extract_pdf(path):
     doc = pymupdf.open(path)
     text = ""
@@ -46,6 +49,7 @@ def extract_pdf(path):
         text += page.get_text()
     return text
 
+
 def extract_docx(path):
     doc = docx.Document(path)
     data = []
@@ -54,6 +58,7 @@ def extract_docx(path):
     content = '\n\n'.join(data)
     return content
 
+
 def extract_pptx(path):
     prs = Presentation(path)
     text = ""
@@ -63,6 +68,7 @@ def extract_pptx(path):
                 text += shape.text + "\n"
     return text
 
+
 def mode_load(path):
     choice = ""
     file_type = path.split(".")[-1]
@@ -79,13 +85,17 @@ def mode_load(path):
         choice = "doc"
         print(content[:100])
         return choice, content[:5000]
+
+
     elif file_type in ["png", "jpg", "jpeg", "bmp", "tiff", "webp"]:
         content = Image.open(path).convert('RGB')
         choice = "image"
         return choice, content
+
     else:
         raise gr.Error("Oops, unsupported files.")
 
+
 @spaces.GPU()
 def stream_chat(message, history: list, temperature: float, max_length: int, top_p: float, top_k: int, penalty: float):
     
@@ -100,9 +110,8 @@ def stream_chat(message, history: list, temperature: float, max_length: int, top
     print(f'history is - {history}')
     conversation = []
     prompt_files = []
-    
-    if "files" in message and message["files"]:
-        choice, contents = mode_load(message["files"][-1].path)
+    if message["files"]:
+        choice, contents = mode_load(message["files"][-1])
         if choice == "image":
             conversation.append({"role": "user", "image": contents, "content": message['text']})
         elif choice == "doc":
@@ -110,9 +119,11 @@ def stream_chat(message, history: list, temperature: float, max_length: int, top
             conversation.append({"role": "user", "content": format_msg})
     else:
         if len(history) == 0:
+            # raise gr.Error("Please upload an image first.")
             contents = None
             conversation.append({"role": "user", "content": message['text']})
         else:
+            # image = Image.open(history[0][0][0])
             for prompt, answer in history:
                 if answer is None:
                     prompt_files.append(prompt[0])
@@ -120,17 +131,17 @@ def stream_chat(message, history: list, temperature: float, max_length: int, top
                 else:
                     conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
             if len(prompt_files) > 0:
-                choice, contents = mode_load(prompt_files[-1].path)
+                choice, contents = mode_load(prompt_files[-1])
             else:
                 choice = ""
                 conversation.append({"role": "user", "image": "", "content": message['text']})
 
+
             if choice == "image":
                 conversation.append({"role": "user", "image": contents, "content": message['text']})
             elif choice == "doc":
                 format_msg = contents + "\n\n\n" + "{} files uploaded.\n" + message['text']
                 conversation.append({"role": "user", "content": format_msg})
-
     print(f"Conversation is -\n{conversation}")
 
     input_ids = tokenizer.apply_chat_template(conversation, tokenize=True, add_generation_prompt=True,
@@ -157,14 +168,19 @@ def stream_chat(message, history: list, temperature: float, max_length: int, top
             buffer += new_text
             yield buffer
 
-chatbot = gr.Chatbot()
 
+chatbot = gr.Chatbot(
+    #rtl=True,
+)
 chat_input = gr.MultimodalTextbox(
     interactive=True,
     placeholder="Enter message or upload a file ...",
     show_label=False,
-)
+    #rtl=True,
+
 
+
+)
 EXAMPLES = [
     [{"text": "Write a poem about spring season in French Language", }],
     [{"text": "what does this chart mean?", "files": ["sales.png"]}],
@@ -178,6 +194,8 @@ with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo:
     gr.ChatInterface(
         fn=stream_chat,
         multimodal=True,
+
+
         textbox=chat_input,
         chatbot=chatbot,
         fill_height=True,
@@ -228,4 +246,5 @@ with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo:
     gr.Examples(EXAMPLES, [chat_input])
 
 if __name__ == "__main__":
-    demo.queue(api_open=False).launch(show_api=False, share=False)
+    
+    demo.queue(api_open=False).launch(show_api=False, share=False, )#server_name="0.0.0.0", )