Update app.py

app.py

@@ -61,6 +61,7 @@ def run_gpt(
     stop_tokens,
     max_tokens,
     seed,
+    purpose,
     **prompt_kwargs,
 ):
     print(seed)
@@ -77,7 +78,7 @@ def run_gpt(
     
     content = PREFIX.format(
         timestamp=timestamp,
-        purpose="Compile the provided data"
+        purpose=purpose
     ) + prompt_template.format(**prompt_kwargs)
     if VERBOSE:
         print(LOG_PROMPT.format(content))
@@ -96,7 +97,7 @@ def run_gpt(
         print(LOG_RESPONSE.format(resp))
     return resp
 
-def compress_data(c, history):
+def compress_data(c, instruct, history):
     seed=random.randint(1,1000000000)
     
     print (c)
@@ -125,7 +126,7 @@ def compress_data(c, history):
             stop_tokens=["observation:", "task:", "action:", "thought:"],
             max_tokens=4096,
             seed=seed,
-            purpose="Compress data",
+            purpose=instruct,
             knowledge=new_history,
             history=hist,
         )
@@ -145,12 +146,13 @@ def compress_data(c, history):
         history="All data has been recieved.",
     )
     print ("final" + resp)
-    history = "observation: {}\n".format(resp)
-    return history
+    #history = "observation: {}\n".format(resp)
+    return resp
 
 
 
-def summarize(inp,history,file=None):
+def summarize(inp,history,data=None,file=None):
+    #file = None
     history = [(inp,"Working on it...")] if not history else history
     
     yield "",history
@@ -164,7 +166,7 @@ def summarize(inp,history,file=None):
             print (e) 
         
     print(inp)
-    out = str(inp)
+    out = str(data)
     rl = len(out)
     print(f'rl:: {rl}')
     c=0
@@ -173,7 +175,7 @@ def summarize(inp,history,file=None):
             c +=1
     print (f'c:: {c}')
     
-    rawp = compress_data(c,out)    
+    rawp = compress_data(c,inp,out)    
     
     #print (rawp)
     #print (f'out:: {out}')