Update app.py

app.py

@@ -119,48 +119,19 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         yield 'test1'
         yield data[1]
         yield output
-        
-        #yield output
-        #yield history
-        #yield prompt
-    #with open(f'{loc_folder}/{loc_file}.json','w') as jobj:
-    #    lod.append({'prompt':prompt,'response':output,'image':im_box,'model':clients[1]['name'],'seed':seed}),
-    #    jobj.write(json.dumps(lod,indent=4))
-    #jobj.close()    
-    #chat_im_out=chat_img(output)
 
-'''def gen_im(prompt,seed):
-    print('generating image')
-    image_out = im_client.text_to_image(prompt=prompt['text'],height=128,width=128,num_inference_steps=10,seed=seed)
-    #print(type(image_out))
-    output=f'images/{uuid.uuid4()}.png'
-    image_out.save(output)
-    print('Done: ',output)
-    return [{'role':'assistant','content': {'path':output}}]'''
 def parse_json(inp):
     print("PARSE INPUT")
     print(inp)
     if type(inp)==type(""):
         lines=""
         if "```" in inp:
-            start = inp.find("```json") + 7  # Find index after the start character
-            end = inp.find("```", start)  # Find index of end character from the start index
+            start = inp.find("```json") + 7  
+            end = inp.find("```", start) 
             if start >= 0 and end >= 0:
-                inp= inp[start:end]  # Slice the string between start and end
+                inp= inp[start:end] 
             else:
-                inp="NONE"  # Return None if characters not found
-            '''tog=False
-            cnt=0
-            for line in inp.split("\n"):
-                if tog==True:
-                    if not "```" in line:
-                        lines+=line
-                if "```" in line:
-                    cnt+=1
-                    if cnt==1:
-                        tog = True
-                    elif cnt==2:
-                        tog = False'''
+                inp="NONE" 
             print("Extracted Lines")
             print(inp)
         try:
@@ -189,12 +160,6 @@ def build_space(repo_name,file_name,file_content,access_token=""):
                 private=False,
             )
         
-        # Create a new Space
-        #response = api.create_repo(repo_path)
-        #space_info = repo_url.json()
-        #print(space_info)
-        #space_id = space_info["name"]
-        #print(f"Created Space with ID: {space_id}")
         local_file_path=str(uuid.uuid4())
         with open(local_file_path, 'w') as f:
             f.write(str(file_content))