Update app.py

app.py

@@ -25,6 +25,7 @@ def predict(inputs, top_p, temperature, openai_api_key, chat_counter, chatbot=[]
         "max_tokens": 1000
     }
 
+
     response = requests.post(API_URL, headers=headers, json=payload, stream=True)
 
     partial_words = ""
@@ -35,23 +36,18 @@ def predict(inputs, top_p, temperature, openai_api_key, chat_counter, chatbot=[]
             if chunk:
                 try:
                     chunk_text = chunk.decode()
-                    print("Raw Chunk:", chunk_text)  # 원시 응답 출력
-                    chunk_data = json.loads(chunk_text[6:])  # JSON 파싱
-                    print("Parsed Data:", chunk_data)  # 파싱된 데이터 출력
-
-                    if 'choices' in chunk_data and 'content' in chunk_data['choices'][0]['delta']:
-                        partial_words += chunk_data['choices'][0]['delta']['content']
-                        if token_counter == 0:
-                            history.append(" " + partial_words)
-                        else:
-                            history[-1] = partial_words
-                        token_counter += 1
-                        chat = [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)]
-                        yield chat, history, chat_counter
+                    # JSON 형식이 올바른지 확인
+                    if chunk_text.strip().startswith("{"):
+                        chunk_data = json.loads(chunk_text)  # JSON 파싱
+                        # ... 나머지 코드 ...
+                    else:
+                        print("Unexpected format:", chunk_text)
                 except json.JSONDecodeError as e:
                     print("JSON 파싱 오류:", e)
+                except Exception as e:
+                    print("응답 처리 오류:", e)
     except Exception as e:
-        print("응답 처리 오류:", e)
+        print("전체 응답 처리 오류:", e)
 
     return chatbot, history, chat_counter